robcaulk dcf9bbdaea ensure data is on same device as the model		2023-04-13 12:19:34 +02:00
..
__init__.py	organize pytorch files	2023-03-21 15:09:54 +01:00
PyTorchDataConvertor.py	fix pytorch data convertor type hints	2023-04-03 19:02:07 +03:00
PyTorchMLPModel.py	Update torch import	2023-04-11 19:38:24 +02:00
PyTorchModelTrainer.py	ensure data is on same device as the model	2023-04-13 12:19:34 +02:00
PyTorchTrainerInterface.py	fix model loading from disk bug, improve doc, clarify installation/docker instructions, add a torch tag to the freqairl docker image. Fix seriously outdated prediction_model docstrings	2023-04-08 12:09:53 +02:00