MLXID/DeepLearning
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Minor bug fix

Browse Source
This commit is contained in:
xie_x1
2026-06-11 16:22:56 +02:00
parent ae7c7a94d6
commit 70f6acc6f8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Train_2Photon.py
+2 -2
View File
@@ -268,6 +268,6 @@ if __name__ == "__main__":
print(f"Saved model checkpoint: {modelName}_E{epoch}.pth")
plot_loss_curves(TrainLosses, ValLosses, test_loss=-1, exp_name=exp_name, conf=conf)
plot_rms_curve(train_rms_xs, train_rms_ys, val_rms_xs, val_rms_ys, test_rms_x = -1, test_rms_y = -1, exp_name=exp_name, conf=conf)
test_loss = evaluate(model, testLoader, loss_fn)
test_loss, rms_2d_test = evaluate(model, testLoader, loss_fn)
plot_loss_curves(TrainLosses, ValLosses, test_loss=test_loss, exp_name=exp_name, conf=conf)
plot_rms_curve(train_rms_xs, train_rms_ys, val_rms_xs, val_rms_ys, test_rms_x = -1, test_rms_y = -1, exp_name=exp_name, conf=conf)
plot_rms_curve(train_rms_xs, train_rms_ys, val_rms_xs, val_rms_ys, test_rms_x = rms_2d_test[0], test_rms_y = rms_2d_test[1], exp_name=exp_name, conf=conf)