Ejemplo n.º 1
0
 def __init__(self, opt, model: BaseModel, dataset: BaseDataset):
     self.opt = opt
     self.model = model
     self.device = model.get_device()
     self.dataset = dataset