fix: the orignal wpdc loss

cleardusk · Jan 26, 2021 · a62d20f · a62d20f
1 parent e44cd27
commit a62d20f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wpdc_loss.py b/wpdc_loss.py
@@ -76,7 +76,7 @@ def _calc_weights_resample(self, input_, target_):
         offset[:, -1] = offsetg[:, -1]
 
         weights = torch.zeros_like(input, dtype=torch.float)
-        tmpv = (u_base + w_shp_base @ alpha_shp + w_exp_base @ alpha_exp).view(N, -1, 3).permute(0, 2, 1)
+        tmpv = (u_base + w_shp_base @ alpha_shpg + w_exp_base @ alpha_expg).view(N, -1, 3).permute(0, 2, 1)
 
         tmpv_norm = torch.norm(tmpv, dim=2)
         offset_norm = sqrt(w_shp_base.shape[0] // 3)