29 Commits

Author	SHA1	Message	Date
Christian Clauss	99b4767bf4	Fix typos discovered by codespell	2024-04-18 22:51:25 +02:00
Robert Caulk	18e34632d8	Update PyTorchModelTrainer.py ... The n_epochs should be defined using the `max` not the `min` function.	2024-03-25 09:21:32 +01:00
yinon	23d2bad2a0	pytorch - set n_steps type as optional	2023-08-04 14:33:59 +00:00
yinon	a3c6904fbc	pytorch - naming refactor - max_iters to n_steps	2023-08-04 13:45:21 +00:00
yinon	d17bf6350d	pytorch - trainer - revert load changes	2023-08-04 12:53:20 +00:00
yinon	777d25192c	pytorch - bugfix - explicitly assign tensor to var as .to() is not inplace operation	2023-08-04 12:53:20 +00:00
yinon	836d7b885a	pytorch - trainer - set default usage of n_epochs instead of max_iters	2023-08-04 12:53:19 +00:00
Yinon Polak	d61f512e20	pytorch - trainer - clean code	2023-07-15 14:43:05 +03:00
Yinon Polak	77f1584713	pytorch - trainer - bugfix step tensorboard step usage	2023-07-15 14:37:44 +03:00
Yinon Polak	ffcba45b1b	pytorch - mypy fixes	2023-07-13 21:36:14 +03:00
Yinon Polak	9fb0ce664c	pytorch - ruff fixes	2023-07-13 21:32:46 +03:00
Yinon Polak	5734358d91	pytorch - trainer - add assertion that either n_epochs or max_iters is been set.	2023-07-13 20:59:33 +03:00
Yinon Polak	7d28dad209	pytorch - add n_epochs param to trainer	2023-07-13 20:41:38 +03:00
yinon	588ffeedc1	pytorch - trainer - reomve max_n_eval_batches arg from estimate loss method	2023-07-13 15:40:40 +00:00
yinon	49a7de4ebd	pytorch - trainer - add device arg to load method	2023-07-13 15:39:47 +00:00
yinon	9cb45a3810	pytorch - bugfix - explicitly assign tensor to var as .to() is not inplace operation	2023-07-13 15:37:50 +00:00
robcaulk	8a9b2fc16f	fix merge conflicts with develop	2023-05-14 12:00:03 +00:00
robcaulk	18c1eda09b	remove commented lines	2023-05-13 11:27:36 +00:00
robcaulk	3ae3cc63df	fix bug in continual_learning for PyTorch* models	2023-05-13 11:14:16 +00:00
robcaulk	ffc4d87263	add tensorboard integration to XGBoost and PyTorch et al	2023-05-10 09:48:36 +00:00
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