Subject | Hash | Author | Date (UTC) |
---|---|---|---|
fix error when save | 86a8f467d47a67da0accc963e8baacbaea07ab14 | Thai Thien | 2020-02-27 17:55:46 |
we discard deform 7 and keep deform 5 instead | 91ff78e4a1aef3677b15447b631c2927c9e62a31 | Thai Thien | 2020-02-27 16:22:47 |
reduce image by crop | 530516ff40b744cac79756c97faaed50c8b65cfb | Thai Thien | 2020-02-27 16:19:11 |
try to solve x8/8 dimension mismatch | 6b0616197995b3c95041cde43030245f96f1f81e | Thai Thien | 2020-02-27 16:17:12 |
reduce model size, discard middle deform (kernal size 5) | 00fb477cf699a8f2eed8e861f241c077b2046000 | Thai Thien | 2020-02-27 16:11:51 |
shanghaitech same size density map | 25fc2e9395dd67bb6cc273c7fe6cd64c3b154120 | Thai Thien | 2020-02-27 16:03:05 |
fix target1.unsqueeze(0) | bb5c46f121207fd9838ee085314df40513e2ed32 | Thai Thien | 2020-02-27 15:56:07 |
because GPU too small, we crop shanghaitech A | 0f97517dd8de5d3d75cf49af9aa9d20e4595ea3e | Thai Thien | 2020-02-27 15:52:21 |
meow | 3b9576511e597b4894c7c95ca06dc5219ef9c524 | tthien | 2020-02-27 15:44:39 |
fix wrong py name in script, dataset use only 1 worker | 3bd765c9aa79e47049a94547e57deb68c947001c | Thai Thien | 2020-02-27 15:41:44 |
minor fix | 5505b9a8b28df105e11a99d3462323ae9d507102 | Thai Thien | 2020-02-27 15:28:37 |
get ready for short training run with 30 epochs | 66dda0858561897cd5f81e10077459adb39d86dd | Thai Thien | 2020-02-27 15:22:01 |
implement attn_can_adcrowdnet | ffd38664a43d861c20cdc225746b9ce2a00260c7 | Thai Thien | 2020-02-27 15:10:27 |
WIP: add can-adcrowdnet | 5620b83449b31d00a367c8de77e431e19a5ccfb3 | Thai Thien | 2020-02-25 11:31:38 |
add readable timestamp viz | ae1fdb49ddb9ea77659529dceb7fb87c2790c8dc | Thai Thien | 2020-02-24 03:49:57 |
change save name prefix | c53a86f30fb8fd4e8f3a409eb67827d56a43ae5c | Thai Thien | 2020-02-02 10:48:15 |
training flow that work | fb242273e8f696916f9d1ff4bb76b4e5869799ef | Thai Thien | 2020-02-02 10:42:01 |
fix the dataloader for shanghaitech | 5f2aee9f316e6555e6a70c6ad037a4e6b491867b | Thai Thien | 2020-02-02 09:19:50 |
context aware visualize seem ok | 1bdb6ffe77ca4e40ef8f299b2506df2266243db4 | Thai Thien | 2020-02-02 05:07:10 |
visualize eval context aware network seem ok | f3fe45c23dfeab3730624737efabb0b14d23c25b | Thai Thien | 2020-02-02 04:50:34 |
File | Lines added | Lines deleted |
---|---|---|
train_attn_can_adcrowdnet.py | 2 | 10 |
File train_attn_can_adcrowdnet.py changed (mode: 100644) (index db87802..2706723) | |||
... | ... | if __name__ == "__main__": | |
77 | 77 | print(timestamp + " Validation set Results - Epoch: {} Avg mae: {:.2f} Avg mse: {:.2f} Avg loss: {:.2f}" | print(timestamp + " Validation set Results - Epoch: {} Avg mae: {:.2f} Avg mse: {:.2f} Avg loss: {:.2f}" |
78 | 78 | .format(trainer.state.epoch, metrics['mae'], metrics['mse'], metrics['nll'])) | .format(trainer.state.epoch, metrics['mae'], metrics['mse'], metrics['nll'])) |
79 | 79 | ||
80 | def score_function(engine): | ||
81 | """ | ||
82 | saver score function | ||
83 | :param engine: | ||
84 | :return: | ||
85 | """ | ||
86 | engine.state.metrics['mae'] | ||
80 | |||
87 | 81 | ||
88 | 82 | # docs on save and load | # docs on save and load |
89 | 83 | to_save = {'trainer': trainer, 'model': model, 'optimizer': optimizer} | to_save = {'trainer': trainer, 'model': model, 'optimizer': optimizer} |
90 | 84 | save_handler = Checkpoint(to_save, DiskSaver('saved_model/' + args.task_id, create_dir=True, atomic=True), | save_handler = Checkpoint(to_save, DiskSaver('saved_model/' + args.task_id, create_dir=True, atomic=True), |
91 | 85 | filename_prefix=args.task_id, | filename_prefix=args.task_id, |
92 | n_saved=5, | ||
93 | score_function=score_function, | ||
94 | score_name="mae") | ||
86 | n_saved=5) | ||
95 | 87 | ||
96 | 88 | trainer.add_event_handler(Events.EPOCH_COMPLETED(every=3), save_handler) | trainer.add_event_handler(Events.EPOCH_COMPLETED(every=3), save_handler) |
97 | 89 |