Skip to content

Commit d6dcf06

Browse files
committed
Add model_path argument in arguments.py and remove saved_model_path
- Introduced a new command-line argument --model_path for specifying the model file path. - Removed the deprecated --saved_model_path argument for clarity and consistency.
1 parent dbd2afa commit d6dcf06

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

fmpose3d/common/arguments.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,14 +73,14 @@ def init(self):
7373
self.parser.add_argument("--reload", action="store_true")
7474
self.parser.add_argument("--model_dir", type=str, default="")
7575
# Optional: load model class from a specific file path
76+
self.parser.add_argument("--model_path", type=str, default="")
7677
self.parser.add_argument("--model_weights_path", type=str, default="")
7778

7879
self.parser.add_argument("--post_refine_reload", action="store_true")
7980
self.parser.add_argument("--checkpoint", type=str, default="")
8081
self.parser.add_argument(
8182
"--previous_dir", type=str, default="./pre_trained_model/pretrained"
8283
)
83-
self.parser.add_argument("--saved_model_path", type=str, default="")
8484

8585
self.parser.add_argument("--n_joints", type=int, default=17)
8686
self.parser.add_argument("--out_joints", type=int, default=17)
@@ -147,7 +147,6 @@ def init(self):
147147
# uncertainty-aware aggregation threshold factor
148148

149149
self.parser.add_argument("--topk", type=int, default=3)
150-
self.parser.add_argument("--weight_softmax_tau", type=float, default=1.0)
151150
self.parser.add_argument("--exp_temp", type=float, default=0.002)
152151
self.parser.add_argument("--mode", type=str, default="exp")
153152

0 commit comments

Comments
 (0)