diff options
Diffstat (limited to 'options/dataset_options.py')
| -rw-r--r-- | options/dataset_options.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/options/dataset_options.py b/options/dataset_options.py index 4b523c7..de1dd57 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -130,7 +130,7 @@ class DatasetOptions(BaseOptions): ) self.parser.add_argument( - '--checkpoint', + '--checkpoint-name', type=str, default='messi', help='which checkpoint to load' |