Commit graph

52 commits

Author SHA1 Message Date
f5656add89 submodule: follow master branch 2024-03-21 18:15:10 +00:00
Mouleesh
7ba869ae4f some changes 2024-03-19 23:06:47 +00:00
Mouleesh
5cbdd4b6aa LMK if want any changes 2024-03-19 22:33:16 +00:00
Mouleesh
ba3e563973 Further filtered the data 2024-03-19 22:29:30 +00:00
Mouleesh
4a914bf551 Code for the graph 2024-03-19 17:16:56 +00:00
Mouleesh
97782aad6e Filtered file graphs 2024-03-19 17:14:05 +00:00
Mouleesh
bb4a4ee0d6 comment 2024-03-13 16:30:34 +00:00
60fc7b7bd1 Added report submodule 2024-03-10 20:40:50 +00:00
100e555233 Added report submodule 2024-03-10 20:38:09 +00:00
3d6aa07983 Merge branch 'main' of https://github.com/rubberhead/mlp-project 2024-03-06 22:40:06 +00:00
Zhengyi Chen
c228fc27cc Edited loss fn to incorporate index of dispersion 2024-03-06 22:37:54 +00:00
Zhengyi Chen
12adb6b7bf Edited loss (ongoing) 2024-03-06 22:30:43 +00:00
Zhengyi Chen
9d2a30a226 Loss revamp & Renamed model to network 2024-03-06 20:44:37 +00:00
0d35d607fe ... 2024-03-06 03:26:28 +00:00
ae9bc34fde FEATURE: export to DataFrame in hdf5 2024-03-06 02:37:18 +00:00
Zhengyi Chen
208091ce8a More training scripts 2024-03-06 01:38:21 +00:00
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
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
a9dd8dee04 Improved preprocess script performance 2024-03-03 19:00:37 +00:00
12aabb0d3f More working than not
Not sure if validation works, call it a day
2024-03-03 03:16:54 +00:00
Zhengyi Chen
4a03211c83 Merge branch 'main' of github.com:rubberhead/mlp-project 2024-03-03 02:00:37 +00:00
Zhengyi Chen
57510503d0 Forgot to commit this 2024-03-03 02:00:03 +00:00
5d77c1da4e Added comment 2024-03-02 23:40:18 +00:00
Zhengyi Chen
c88b938680 Extra changes, also not sure if works 2024-03-02 23:32:19 +00:00
Zhengyi Chen
f059924b75 Crude impl of glue layer, not sure anything works 2024-03-02 23:18:19 +00:00
b69727b74f Added dataset making script 2024-03-02 19:26:55 +00:00
359ed8c579 Test 2024-03-02 17:52:10 +00:00
6456d3878e Added save_checkpoint 2024-02-29 19:04:25 +00:00
ab633da4a5 Added dummy input for one-shot size-checking 2024-02-29 18:44:42 +00:00
Zhengyi Chen
dcc3f57596 Call it a day 2024-02-28 17:26:02 +00:00
04f78fbcbc maintenance 2024-02-28 15:28:38 +00:00
Zhengyi Chen
99266d9c92 Added sth more 2024-02-27 21:27:02 +00:00
49a913a328 Added STNet-prepended TransCrowd GAP 2024-02-27 18:57:13 +00:00
c2a1bb46ef Isolated STNet from test pipeline 2024-02-27 17:55:57 +00:00
b8f7f922f1 FIX: valid_loader instead of test_loader 2024-02-25 21:15:50 +00:00
62df7464e4 lemme cook alright? 2024-02-25 21:03:32 +00:00
b6d2460060 Added comment 2024-02-05 21:51:17 +00:00
8e0e82f67a Added comment 2024-02-05 14:39:00 +00:00
322d7f9ea5 Added transcrowd_gap 2024-02-05 14:01:27 +00:00
bcff06f9c2 Added comments 2024-01-30 17:06:29 +00:00