ArneBinder / pytorch-ie-hydra-template-1

PyTorch-IE Hydra Template
8 stars 1 forks source link

make test_train_eval() deterministic #80

Closed ArneBinder closed 1 year ago

ArneBinder commented 1 year ago

test_train_eval() occasionally fails because the test F1 remains zero. Now, we ensure reproducibility by setting the seed and trainer.deterministic=True. Furthermore, we just check that the test scores are the same as after training (and do not ensure that it is above zero).