Age | Commit message (Collapse) | Author | |
---|---|---|---|
2021-01-11 | Add evaluation script, code review and fix some bugs | Jordan Gong | |
1. Add new `train_all` method for one shot calling 2. Print time used in 1k iterations 3. Correct label dimension in predict function 4. Transpose distance matrix for convenient indexing 5. Sort dictionary before generate signature 6. Extract visible CUDA setting function | |||
2021-01-10 | Make predict function transform samples different conditions in a single shot | Jordan Gong | |
2021-01-07 | Train different models in different conditions | Jordan Gong | |
2021-01-07 | Add typical training script and some bug fixes | Jordan 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 |