Spaces:
Running
on
Zero
Running
on
Zero
Add --logdir argument to sample.py parser and restore it in app.py
Browse files
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
|