9 Commits

Author	SHA1	Message	Date
robcaulk	af139ffbab	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
robcaulk	dcf9bbdaea	ensure data is on same device as the model	2023-04-13 12:19:34 +02:00
robcaulk	48d3c8e62e	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
Yinon Polak	d9d9993179	add documentation	2023-04-03 17:06:39 +03:00
Yinon Polak	c137666230	fix imports	2023-04-03 16:03:15 +03:00
Yinon Polak	bd3b70293f	add pytorch data convertor	2023-04-03 15:19:10 +03:00
Yinon Polak	077a947972	clean code	2023-03-28 15:18:10 +03:00
Yinon Polak	026b6a39a9	bugfix skip test split when empty	2023-03-28 14:40:23 +03:00
robcaulk	1ba01746a0	organize pytorch files	2023-03-21 15:09:54 +01:00