更新 models/rome_unsb_model.py

2025-02-23 15:27:15 +08:00 · 2025-02-23 15:27:15 +08:00 · 40d33fd9c2
commit 40d33fd9c2
parent 24b8f9acae
1 changed files with 1 additions and 1 deletions
--- a/models/rome_unsb_model.py
+++ b/models/rome_unsb_model.py
@ -190,7 +190,7 @@ class ContentAwareTemporalNorm(nn.Module):
        
        return F_content

-class CTNxModel(BaseModel):
+class RomaUnsbModel(BaseModel):
    @staticmethod
    def modify_commandline_options(parser, is_train=True):
        """配置 CTNx 模型的特定选项"""