def __init__(self, params):
self.device_id = params["device_id"]
self.epochs = params["epochs"]
# data
self.train_data = Data(params["train_data"])
#self.test_data = Data(params["test_data"])
#self.validation_data = Data(params["validation_data"])
#self.progress_train_data = Data(params["progress_train_data"])
self.progress_train_data = None
self.progress_test_data = Data(params["progress_test_data"])
# model
self.model = Model.get(params["model"])
print(self.model)
if self.device_id != -1:
self.model = self.model.cuda(self.device_id)
# optimizer
self.optimizer = Optimizer.get(self.model, params["optimizer"])
# loss
self.loss_func = Loss.get(params["loss"])
#if self.device_id != -1:
# self.loss_func = self.loss_func.cuda(self.device_id)
# progress evaluator
self.progress_evaluator = ProgressEvaluator.get(
params["progress_evaluator"], self.progress_train_data,
self.progress_test_data, self.device_id)
