From 40d33fd9c296d2967bc94f4953095a4bd72d9a9b Mon Sep 17 00:00:00 2001
From: 123456 <3351416005@qq.com>
Date: Sun, 23 Feb 2025 15:27:15 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20models/rome=5Funsb=5Fmodel?=
 =?UTF-8?q?.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 models/rome_unsb_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 模型的特定选项"""