Skip to content

Commit 8b402c3

Browse files
committed
Refactor FMPose3D_main.py: move import of resolve_weights_path to the top for better organization and remove redundant import in the main execution block.
1 parent 1eb00b4 commit 8b402c3

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

scripts/FMPose3D_main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
import torch.optim as optim
1919
from tqdm import tqdm
2020

21+
from fmpose3d.utils.weights import resolve_weights_path
2122
from fmpose3d.common import opts, Human36mDataset, Fusion
2223
from fmpose3d.common.utils import *
2324

@@ -341,7 +342,6 @@ def print_error_action(action_error_sum, is_train):
341342

342343
if args.reload:
343344
model_dict = model["CFM"].state_dict()
344-
from fmpose3d.utils.weights import resolve_weights_path
345345
model_path = resolve_weights_path(args.model_weights_path, args.model_type)
346346

347347
print(f"Loading weights from: {model_path}")

0 commit comments

Comments
 (0)