--- a +++ b/tasks_base/generate.py @@ -0,0 +1,26 @@ +'''Training script generator for subject-independent classification. +''' +with open('script.sh.template', 'r') as f: + sh_template = f.read() +with open('train.pbs.template', 'r') as f: + pbs_template = f.read() + + +def list_to_str(l): + return ' '.join(map(str, l)) + + +all_folds = list(range(54)) +command_template = "python train_base.py $datapath $outpath -fold {0} -gpu {1} > $logpath/stdout.f{0}.out &" +for i, j in enumerate(range(0, len(all_folds), 8)): + folds = all_folds[j:j+8] + commands = [] + for ind, fold in enumerate(folds): + commands.append(command_template.format(fold, ind)) + commands.append('wait') + content = sh_template + '\n'.join(commands) + with open('script.' + str(i) + '.sh', 'w') as f: + f.write(content) + content = pbs_template.format(i) + with open('train.' + str(i) + '.pbs', 'w') as f: + f.write(content)