revert eefe7f26d480e72a3871b0ba171c20aa58a64127

revert 更新 models/roma_model.py
This commit is contained in:
123456 2025-02-23 15:26:19 +08:00
parent eefe7f26d4
commit 24b8f9acae

View File

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