17 commits

Author	SHA1	Message	Date
Zhengyi Chen	733b3c5387	Disable img writer	2024-03-05 21:27:27 +00:00
Zhengyi Chen	524ee03187	TEST: use tensorboard for stuff	2024-03-04 20:32:26 +00:00
Zhengyi Chen	2d31162c58	TEST: use autocast for mixed-precision training	2024-03-04 18:36:40 +00:00
Zhengyi Chen	94867bd8bf	Use mse instead of l1 for info retainment loss, fix mae	2024-03-04 14:39:58 +00:00
Zhengyi Chen	d46a027e3f	Updated glue impl, complex loss fn	2024-03-04 04:10:17 +00:00
rubberhead	83fcc43f0b	TEST: train on gt_count instead of kpoint	2024-03-04 01:49:09 +00:00
Zhengyi Chen	ee50e84946	FIX: vaidation code	2024-03-03 23:56:17 +00:00
Zhengyi Chen	c74f4c7fb3	Sync	2024-03-03 23:13:57 +00:00
rubberhead	da8287b7e8	FIX: deprecated loss reduction argument	2024-03-03 22:39:30 +00:00
Zhengyi Chen	c4905acf6d	Sync	2024-03-03 22:19:47 +00:00
Zhengyi Chen	fc941ebaf7	Yay, works in DP via CPU	2024-03-03 21:59:58 +00:00
Zhengyi Chen	ab15419d2f	Debug	2024-03-03 19:40:22 +00:00
rubberhead	12aabb0d3f	More working than not ... Not sure if validation works, call it a day	2024-03-03 03:16:54 +00:00
rubberhead	359ed8c579	Test	2024-03-02 17:52:10 +00:00
rubberhead	6456d3878e	Added save_checkpoint	2024-02-29 19:04:25 +00:00
Zhengyi Chen	dcc3f57596	Call it a day	2024-02-28 17:26:02 +00:00
Zhengyi Chen	99266d9c92	Added sth more	2024-02-27 21:27:02 +00:00

Renamed from train-revpers.py (Browse further)