Subject | Hash | Author | Date (UTC) |
---|---|---|---|
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 |
mse-ssim loss function | 052de7c879bff7690a7cfc1905c8376bf8605c45 | Thai Thien | 2020-04-30 11:41:24 |
File | Lines added | Lines deleted |
---|---|---|
data_flow.py | 5 | 3 |
File data_flow.py changed (mode: 100644) (index 1ae9805..e05400d) | |||
... | ... | def get_dataloader(train_list, val_list, test_list, dataset_name="shanghaitech", | |
771 | 771 | dataset_name=dataset_name), | dataset_name=dataset_name), |
772 | 772 | batch_size=batch_size, | batch_size=batch_size, |
773 | 773 | num_workers=0, | num_workers=0, |
774 | collate_fn=my_collate) | ||
774 | collate_fn=my_collate, pin_memory=False) | ||
775 | 775 | ||
776 | 776 | if val_list is not None: | if val_list is not None: |
777 | 777 | val_loader = torch.utils.data.DataLoader( | val_loader = torch.utils.data.DataLoader( |
... | ... | def get_dataloader(train_list, val_list, test_list, dataset_name="shanghaitech", | |
781 | 781 | train=False, | train=False, |
782 | 782 | dataset_name=dataset_name), | dataset_name=dataset_name), |
783 | 783 | num_workers=0, | num_workers=0, |
784 | batch_size=1) | ||
784 | batch_size=1, | ||
785 | pin_memory=False) | ||
785 | 786 | else: | else: |
786 | 787 | val_loader = None | val_loader = None |
787 | 788 | ||
... | ... | def get_dataloader(train_list, val_list, test_list, dataset_name="shanghaitech", | |
793 | 794 | train=False, | train=False, |
794 | 795 | dataset_name=dataset_name), | dataset_name=dataset_name), |
795 | 796 | num_workers=0, | num_workers=0, |
796 | batch_size=1) | ||
797 | batch_size=1, | ||
798 | pin_memory=False) | ||
797 | 799 | else: | else: |
798 | 800 | test_loader = None | test_loader = None |
799 | 801 |