PROTAC-Degradation-Predictor / tests /test_pytorch_model.py

Commit History

Working loading and predicting with pretrained model + Added scaling in "pure" pytorch during forwarding
15216c3

ribesstefano commited on

Added model tests + checkpointing of the scaler object
de956c8

ribesstefano commited on