Subject | Hash | Author | Date (UTC) |
---|---|---|---|
torch.set_num_threads(4) | a45027f9b9088eb55da8fa7994dd6944e572b0e2 | Thai Thien | 2020-04-30 19:33:50 |
pin_memory=False | dabbd7134c597e184c1be88831f15416579ceddc | Thai Thien | 2020-04-30 19:31:17 |
debug cpu consume | d5d9699e2fed3725cd5467cdacd1b5762f2b988f | Thai Thien | 2020-04-30 19:20:38 |
only import ssim when it is needed | 3088d6fe240f94cb47057d76e9fc4da48f2d3aaf | Thai Thien | 2020-04-30 19:16:10 |
no more batch timer | 3fea48c4642f62ab92accbe2493b6275d68c1b02 | Thai Thien | 2020-04-30 19:04:16 |
num_worker=0 | 9e21a25fd4194e99c7899f7c6b6f04d51b432c61 | Thai Thien | 2020-04-30 19:01:09 |
.to(device) | 136bafc672194946e944e85d43880f3342abf9be | Thai Thien | 2020-04-30 18:52:05 |
we load, so increase epoch | b147d3d824c0597390900cb3506495ad9d842bd4 | Thai Thien | 2020-04-30 18:45:20 |
wrong note | 31e67446689b4cecb8f118c43e6794fa437e8eb5 | Thai Thien | 2020-04-30 18:44:38 |
use ssim ccnn | 311308b98fc0a426948953ecdf8b9eec6dfd69dd | Thai Thien | 2020-04-30 18:38:25 |
loss reduce as we train, it might ok | cf1b359106f6e707e6862f2b3cbc9b5c85466ea4 | Thai Thien | 2020-04-30 18:34:19 |
replace SSIM with kornia SSIM | 1f29782fe73751182b696c18567c1de3ab84a6bc | Thai Thien | 2020-04-30 17:13:09 |
fix link | 3c55cee5cbc9fbaee0859b6ab61529029caee681 | Thai Thien | 2020-04-30 15:41:10 |
update documentation | feaaa4095d89e99f71f0684eec302d01e421708d | Thai Thien | 2020-04-30 15:35:38 |
change gpu | c7c6aa29ccee5736c2d4cb290c2ff71c6d897d57 | Thai Thien | 2020-04-30 12:31:11 |
reduce batch size | 2b2820095211e117c6d02236ef85adb89b70098f | Thai Thien | 2020-04-30 12:28:32 |
forget .to(device) | a641d5cdf82eb89c9034253785e80dd920c1f5d6 | Thai Thien | 2020-04-30 12:25:22 |
reduce batch size of shb keepfull | 044f6785fedcf442def0fdb24109413a79274289 | Thai Thien | 2020-04-30 12:20:21 |
ccnn v7 with ssim loss | 40079e5354eef95e0e89cf23a7f2025ee362e232 | Thai Thien | 2020-04-30 12:16:19 |
local | 1d4e08754bd493e76cf09076de7c76adf1a30a0a | Thai Thien | 2020-04-30 12:10:15 |
File | Lines added | Lines deleted |
---|---|---|
experiment_meow_main.py | 2 | 1 |
File experiment_meow_main.py changed (mode: 100644) (index 4a932f8..6353cbe) | |||
... | ... | def very_simple_param_count(model): | |
28 | 28 | ||
29 | 29 | ||
30 | 30 | if __name__ == "__main__": | if __name__ == "__main__": |
31 | torch.set_num_threads(4) # 4 thread | ||
31 | 32 | experiment = Experiment(project_name=PROJECT_NAME, api_key=COMET_ML_API) | experiment = Experiment(project_name=PROJECT_NAME, api_key=COMET_ML_API) |
32 | 33 | device = torch.device("cuda" if torch.cuda.is_available() else "cpu") | device = torch.device("cuda" if torch.cuda.is_available() else "cpu") |
33 | 34 | print(device) | print(device) |
... | ... | if __name__ == "__main__": | |
191 | 192 | ||
192 | 193 | def checkpoint_valid_mae_score_function(engine): | def checkpoint_valid_mae_score_function(engine): |
193 | 194 | score = engine.state.metrics['mae'] | score = engine.state.metrics['mae'] |
194 | return score | ||
195 | return -score | ||
195 | 196 | ||
196 | 197 | ||
197 | 198 | # docs on save and load | # docs on save and load |