Skip to content

Commit 1a60c74

Browse files
committed
fix(option): The parameter data was previously hardcoded as a required input in the training loop. This update allows users to specify data, even if it won't be utilized.
1 parent 54f0d5f commit 1a60c74

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

deepem/train/option.py

+1
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ def initialize(self):
1919
self.parser.add_argument('--exp_name', required=True)
2020
self.parser.add_argument('--model', required=True)
2121
self.parser.add_argument('--sampler', required=True)
22+
self.parser.add_argument('--data', default=None)
2223
self.parser.add_argument('--augment', default=None)
2324
self.parser.add_argument('--modifier', default=None)
2425
self.parser.add_argument('--modifier_kwargs', type=json.loads, default={})

0 commit comments

Comments
 (0)