You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I noticed that the training code seems to select the best model using the test set, since the dataset class only uses a boolean flag training (True for training data, False for test data).
For a fair evaluation, it would be better to use a separate validation set for model selection instead of relying on the test set. This ensures that the reported test performance truly reflects generalization, without being indirectly optimized during training.