yinon 49a7de4ebd pytorch - trainer - add device arg to load method		2023-07-13 15:39:47 +00:00
..
__init__.py	organize pytorch files	2023-03-21 15:09:54 +01:00
datasets.py	add transformer with positional encoding, fix some odds and ends in pytorch, upgrade to PyTorch 2.0	2023-05-01 13:18:03 +00:00
PyTorchDataConvertor.py	pytorch - data convertor - create tensor directly on device, simplify code	2023-07-13 15:38:58 +00:00
PyTorchMLPModel.py	add transformer with positional encoding, fix some odds and ends in pytorch, upgrade to PyTorch 2.0	2023-05-01 13:18:03 +00:00
PyTorchModelTrainer.py	pytorch - trainer - add device arg to load method	2023-07-13 15:39:47 +00: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
PyTorchTransformerModel.py	Fix odd import	2023-05-21 09:50:59 +02:00