Skip to content

Commit af6ec60

Browse files
committed
Update model path variable in vis_in_the_wild.py to align with recent refactoring
- Changed model_path from args.saved_model_path to args.model_weights_path for consistency with other updates.
1 parent 6608049 commit af6ec60

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

demo/vis_in_the_wild.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,7 @@ def get_pose3D(path, output_dir, type='image'):
285285

286286
# if args.reload:
287287
model_dict = model['CFM'].state_dict()
288-
model_path = args.saved_model_path
288+
model_path = args.model_weights_path
289289
print(model_path)
290290
pre_dict = torch.load(model_path)
291291
for name, key in model_dict.items():

0 commit comments

Comments
 (0)