From 4a284084c253b9114fc02e1782962556ff113761 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Thu, 7 Jan 2021 18:37:43 +0800
Subject: Add typical training script and some bug fixes

1. Resolve deprecated scheduler stepping issue
2. Make losses in the same scale(replace mean with sum in separate triplet loss, enlarge pose similarity loss 10x)
3. Add ReLU when compute distance in triplet loss
4. Remove classes except Model from `models` package init
---
 test/cuda.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'test/cuda.py')

diff --git a/test/cuda.py b/test/cuda.py
index ef0ea36..b1418c4 100644
--- a/test/cuda.py
+++ b/test/cuda.py
@@ -1,6 +1,6 @@
 import torch
 
-from models import RGBPartNet
+from models.rgb_part_net import RGBPartNet
 
 P, K = 2, 4
 N, T, C, H, W = P * K, 10, 3, 64, 32
-- 
cgit v1.2.3