Skip to content

Commit 16c5776

Browse files
author
Fangchang Ma
committed
minor change to output folder naming
1 parent ed750c2 commit 16c5776

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,9 +87,9 @@ def adjust_learning_rate(optimizer, epoch, lr_init):
8787

8888
def get_output_directory(args):
8989
output_directory = os.path.join('results',
90-
'{}.sparsifier={}.modality={}.arch={}.decoder={}.criterion={}.lr={}.bs={}.pretrained={}'.
91-
format(args.data, args.sparsifier, args.modality, args.arch, \
92-
args.decoder, args.criterion, args.lr, args.batch_size, \
90+
'{}.sparsifier={}.samples={}.modality={}.arch={}.decoder={}.criterion={}.lr={}.bs={}.pretrained={}'.
91+
format(args.data, args.sparsifier, args.num_samples, args.modality, \
92+
args.arch, args.decoder, args.criterion, args.lr, args.batch_size, \
9393
args.pretrained))
9494
return output_directory
9595

0 commit comments

Comments
 (0)