leoeric commited on
Commit
a1fd3d3
·
1 Parent(s): cc626cb

Add --logdir argument to sample.py parser and restore it in app.py

Browse files
Files changed (1) hide show
  1. sample.py +1 -0
sample.py CHANGED
@@ -342,6 +342,7 @@ if __name__ == '__main__':
342
  # Model config
343
  parser.add_argument('--model_config_path', required=True, type=str, help='path to YAML config file or directory containing config file')
344
  parser.add_argument('--checkpoint_path', required=True, type=str, help='path to local checkpoint file (required when using model_config_path)')
 
345
  parser.add_argument('--save_folder', default=0, type=int)
346
 
347
  # Caption, condition
 
342
  # Model config
343
  parser.add_argument('--model_config_path', required=True, type=str, help='path to YAML config file or directory containing config file')
344
  parser.add_argument('--checkpoint_path', required=True, type=str, help='path to local checkpoint file (required when using model_config_path)')
345
+ parser.add_argument('--logdir', default='./logs', type=pathlib.Path, help='output directory for generated samples')
346
  parser.add_argument('--save_folder', default=0, type=int)
347
 
348
  # Caption, condition