32 Commits

Author	SHA1	Message	Date
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	0c9aa86885	pytorch - data convertor - create tensor directly on device, simplify code	2023-07-13 15:38:58 +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
Matthias	a87b215d67	Fix odd import	2023-05-21 09:50:59 +02: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	36e1e58dad	fix arch	2023-05-06 17:40:04 +00:00
robcaulk	3bbb7e38ea	improve transformer architecture, remove 3.10 install constraint, add documentation for torch.compile()	2023-05-06 16:12:10 +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
Matthias	9857675a5e	Update torch import	2023-04-11 19:38:24 +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	a655524221	pytorch mlp rename input to fix mypy error	2023-04-04 12:24:29 +03:00
Yinon Polak	26738370c7	pytorch mlp add explicit annotation to fix mypy error	2023-04-04 12:12:02 +03:00
Yinon Polak	6b204c97ed	fix pytorch data convertor type hints	2023-04-03 19:02:07 +03:00
Yinon Polak	0c4574b3b7	prevent mypy error, explicitly unpack input list of pytorch mlp model,	2023-04-03 18:10:47 +03:00
Yinon Polak	d9d9993179	add documentation	2023-04-03 17:06:39 +03:00
Yinon Polak	7b494c8333	add documentation to pytorch data convertor	2023-04-03 16:39:49 +03:00
Yinon Polak	bc9454e0f9	add device to data convertor class doc	2023-04-03 16:36:38 +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