Subject | Hash | Author | Date (UTC) |
---|---|---|---|
MSEL1Loss | e2f4dc7bb1bee096ae17101331e0f221d05ba1af | Thai Thien | 2020-07-13 09:34:11 |
t10 | 2f443c077e186901cdbe2c9faf27d2f37905a775 | Thai Thien | 2020-07-12 18:57:39 |
t10 | 87d6dd6e7debf45e16953e39471c13071039fa59 | Thai Thien | 2020-07-12 18:57:17 |
fix | 45617766d305df841b7b0af3c2f6204a2e6d4f46 | Thai Thien | 2020-07-12 18:45:01 |
ccnn_v7_t8_shb | 9f921d9415b89d9c9d4f4e8ef9b65b32663e06ec | Thai Thien | 2020-07-12 16:00:42 |
fix shanghaitech_non_overlap | 9332377476d2d7524822ba98e2321c75be90f709 | Thai Thien | 2020-07-12 15:57:08 |
fix flatten collate | bb25738b0510915712a17daeb590b668b446b0be | Thai Thien | 2020-07-12 15:17:14 |
ccnn_v7_t6_shb | c4f59ee01cbdab3f506302588b9667cd1c9f6411 | Thai Thien | 2020-07-12 14:19:03 |
do not *4 root if we have flatten augmentation list | c534aa36bf314ea32643e92231194bd020d7bf1f | Thai Thien | 2020-07-12 14:19:00 |
train val split shb | 61581543d16aaa2640bdee0b3573e41d1843770d | Thai Thien | 2020-07-12 14:06:25 |
flatten collate | c04708ae0defc81dbf441395e1d27de6a1d598fc | Thai Thien | 2020-07-12 13:54:01 |
travis remove nightly | d4b0c714823046bbafcd3c816d56f7079c76d126 | Thai Thien | 2020-07-12 13:27:55 |
travis | e6368ec3102e01f1bdc71a80a78f0db3617d7e08 | Thai Thien | 2020-07-12 12:33:21 |
flatten_collate | 1e460396875c205c42de27449f56e73cd4ec10e0 | Thai Thien | 2020-07-12 12:23:40 |
train val split test ratio to 0.1 | 5091da3f0b45d875a38c2829e4fec5e61116e869 | Thai Thien | 2020-07-11 03:14:26 |
f | 1defee5dc452c2da5fb540ff1050f6e01fe1878b | Thai Thien | 2020-07-10 17:04:58 |
fix log | db4d655a313b8f3951baf225a4b197fce4bcdd4b | Thai Thien | 2020-07-10 16:50:03 |
typo, trunc 3.0 not 4.0 | ee29c49efd0a5087f11662997b8992d25671a33a | Thai Thien | 2020-07-10 16:44:55 |
let try with trunc 3.0 | f8eac179b1fa79d6451f19a9e6a35b82b94646a4 | Thai Thien | 2020-07-10 16:43:41 |
train script | 7e6984a9eaa6609182ad6c786f6742dd29f1d017 | Thai Thien | 2020-06-14 10:23:05 |
File | Lines added | Lines deleted |
---|---|---|
MSEL1Loss.py | 14 | 0 |
File MSEL1Loss.py added (mode: 100644) (index 0000000..2da7055) | |||
1 | import torch | ||
2 | import torch.functional as F | ||
3 | from torch.nn.modules.loss import _Loss | ||
4 | |||
5 | |||
6 | class MSEL1Loss(_Loss): | ||
7 | |||
8 | __constants__ = ['reduction'] | ||
9 | |||
10 | def __init__(self, size_average=None, reduce=None, reduction='mean'): | ||
11 | super(MSEL1Loss, self).__init__(size_average, reduce, reduction) | ||
12 | |||
13 | def forward(self, input, target): | ||
14 | return F.mse_loss(input, target, reduction=self.reduction) + F.l1_loss(input, target, reduction=self.reduction) |