from traindata import loss_train_hist, acc_train_hist, loss_val_hist = []
acc_val_hist = []