Commit graph

12 commits

Author SHA1 Message Date
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
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
12aabb0d3f More working than not
Not sure if validation works, call it a day
2024-03-03 03:16:54 +00:00
359ed8c579 Test 2024-03-02 17:52:10 +00:00
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)