diff --git a/models/rome_unsb_model.py b/models/rome_unsb_model.py
index 64a67f1..046cbdd 100644
--- 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 模型的特定选项"""