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

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

diff --git a/models/roma_model.py b/models/roma_model.py
index 12536b9..7071296 100644
--- a/models/roma_model.py
+++ b/models/roma_model.py
@@ -15,7 +15,7 @@ import math
 
 from torchvision.transforms import transforms as tfs
 
-class ROMAModel(BaseModel):
+class ROMAUNSBModel(BaseModel):
     
     @staticmethod
     def modify_commandline_options(parser, is_train=True):