更新 models/roma_model.py

This commit is contained in:
123456 2025-02-23 15:25:22 +08:00
parent 515907f2e8
commit eefe7f26d4

View File

@ -15,7 +15,7 @@ import math
from torchvision.transforms import transforms as tfs from torchvision.transforms import transforms as tfs
class ROMAModel(BaseModel): class ROMAUNSBModel(BaseModel):
@staticmethod @staticmethod
def modify_commandline_options(parser, is_train=True): def modify_commandline_options(parser, is_train=True):