Subject | Hash | Author | Date (UTC) |
---|---|---|---|
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 |
delete misinformation note | ca07533d585def04eec086685f2e72eacc330ddc | Thai Thien | 2020-04-29 16:59:47 |
typo | 3505077301af5349665f12121862db2512ad450a | Thai Thien | 2020-04-29 16:59:03 |
change, adapt, survive | 6755c8375302af2146cb63adc967631d53a7b1c8 | Thai Thien | 2020-04-29 16:48:10 |
gpu | 861ec0d41cea2eec359c6ddfe207e1ed6b583369 | Thai Thien | 2020-04-28 18:08:11 |
ccnn_v7_t4 | a84f63e64fe8b31fe22c94d383de5ed4e1a27fe4 | Thai Thien | 2020-04-28 18:07:33 |
2500 | d37516f53d21f3dfc05143ba5ffda80fc5a07825 | Thai Thien | 2020-04-28 17:56:57 |
change max epoch | 246ae30e6dd2b4a15b7dc70a3dc05592ac1c48f2 | Thai Thien | 2020-04-28 17:55:23 |
h1 t3 h1 t4 | fbcd13dd240e06a982a1ce48f27cd1d542a26b63 | Thai Thien | 2020-04-28 17:53:06 |
h1 | 96204cb5262500020371637741131d24e3fa91d0 | Thai Thien | 2020-04-27 17:35:51 |
typo | adfb213c2564bc90b8b69811469534b004808644 | Thai Thien | 2020-04-27 17:17:58 |
batch size 8 for shb | c90fa9a5d725a1ef0d29ed23f947ee05b9aa7894 | Thai Thien | 2020-04-27 17:10:34 |
change proxy | 30cd53782eb17b416c471502f1e6c6e9975a644b | Thai Thien | 2020-04-27 17:06:28 |
experiment with CCNN when we max pooling after merge | c8e8daec89ee71c2f30e30cf3030298ee7073e56 | Thai Thien | 2020-04-27 16:52:50 |
refactor train_compact_cnn | 9d1ecece2382b79a98e6cf2d4579ae68172dbb6a | Thai Thien | 2020-04-27 16:35:47 |
we add head experiment | 82a25e6b89945609486cbafc433eaea20dcdee39 | Thai Thien | 2020-04-27 16:23:50 |
File | Lines added | Lines deleted |
---|---|---|
train_compact_cnn.py | 1 | 1 |
File train_compact_cnn.py changed (mode: 100644) (index 4b570dc..8f81256) | |||
... | ... | if __name__ == "__main__": | |
82 | 82 | ||
83 | 83 | # loss function | # loss function |
84 | 84 | if args.use_ssim: | if args.use_ssim: |
85 | loss_fn = MseSsimLoss() | ||
85 | loss_fn = MseSsimLoss().to(device) | ||
86 | 86 | print("use ssim") | print("use ssim") |
87 | 87 | else: | else: |
88 | 88 | loss_fn = nn.MSELoss(reduction='sum').to(device) | loss_fn = nn.MSELoss(reduction='sum').to(device) |