def __init__(self, opt, model: BaseModel, dataset: BaseDataset): self.opt = opt self.model = model self.device = model.get_device() self.dataset = dataset