chouse_train/tmp_lcfg/lcfgmaker.py

import os

bbnames = ['moblienetv2','resnet50']
dsversion = 201120
dsnames = ['middle', 'box', 'capitallima', 'carexterior', 'carlamp', 'cosmeticbottle', 'slotted']
epoch = 100
saveperiod = 1
lossfunctions = ['CrossEntropyLoss', 'MultiLabelMarginLoss']
batchsizes = ['32','64']
optimizers = ['SGD', 'RMSprop', 'Adam']
learningloss = 0.001


count = 1

for bbname in bbnames:
    for dsname in dsnames:
        for lossfunction in lossfunctions:
            for optimizer in optimizers:
                role = dsname+"_"+ bbname + "_" + str(count) +'_'+ str(dsversion)
                current_dir = os.path.join(os.getcwd(),role)
                f  = open(current_dir + '.lcfg','w')
                if dsname == 'middle':
                    f.write(role +' ' +bbname +' '+ str(dsversion) +' '+ str(dsname) +' '+ str(epoch) +' '+ str(batchsizes[1]) +' '+ optimizer +' '+ str(learningloss) +' '+ lossfunction +' '+ str(saveperiod))
                    f.close()
                else:
                    f.write(role +' ' +bbname +' '+ str(dsversion) +' '+ str(dsname) +' '+ str(epoch) +' '+ str(batchsizes[0]) +' '+ optimizer +' '+ str(learningloss) +' '+ lossfunction +' '+ str(saveperiod))
                    f.close()

                print(current_dir)
                count += 1