Skip to content

Commit 4c8b201

Browse files
committed
update config: replace model_path with model_type from the registry
1 parent 50125e5 commit 4c8b201

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

fmpose3d/common/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ class ModelConfig:
2222
"""Model architecture configuration."""
2323

2424
model: str = ""
25+
model_type: str = "fmpose3d"
2526
layers: int = 3
2627
channel: int = 512
2728
d_hid: int = 1024
@@ -31,7 +32,6 @@ class ModelConfig:
3132
in_channels: int = 2
3233
out_channels: int = 3
3334
frames: int = 1
34-
model_path: str = ""
3535
"""Optional: load model class from a specific file path."""
3636

3737

0 commit comments

Comments
 (0)