Advanced Computing Platform for Theoretical Physics

Commit ed9c90cf authored by Lei Wang's avatar Lei Wang
Browse files

fixe use_density_loss for argparse

parent ac966158
......@@ -40,7 +40,10 @@ echo "CUDA devices $CUDA_VISIBLE_DEVICES"
echo Job started at `date`\n'''
job +='python '+ str(bin) + ' '
for key, val in args.items():
job += '--'+str(key) + ' '+ str(val) + ' '
if key == 'use_density_loss':
if val: job += '--'+str(key) + ' '
else:
job += '--'+str(key) + ' '+ str(val) + ' '
job += '''
echo Job finished at `date`\n'''
......
......@@ -9,7 +9,7 @@ parser.add_argument("--G", type=int, default=300, help="G")
parser.add_argument("--batchsize", type=int, default=16, help="batchsize")
parser.add_argument("--epochs", type=int, default=100000, help="epochs")
parser.add_argument("--lr", type=float, default=1e-4, help="learning rate")
parser.add_argument("--use_density_loss", type=bool, default=False, help="")
parser.add_argument("--use_density_loss", action='store_true', help="")
#model parameters
parser.add_argument("--depth", type=int, default=2, help="depth")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment