summaryrefslogtreecommitdiff
path: root/train.py
AgeCommit message (Collapse)Author
2021-01-07Train different models in different conditionsJordan Gong
2021-01-07Add typical training script and some bug fixesJordan Gong
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