Spaces:
Runtime error
Runtime error
File size: 442 Bytes
1b2a9b1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from swapae.models import MultiGPUModelWrapper
class BaseOptimizer():
@staticmethod
def modify_commandline_options(parser, is_train):
return parser
def __init__(self, model: MultiGPUModelWrapper):
self.opt = model.opt
def train_one_step(self, data_i, total_steps_so_far):
pass
def get_visuals_for_snapshot(self, data_i):
return {}
def save(self, total_steps_so_far):
pass
|