Age | Commit message (Collapse) | Author | |
---|---|---|---|
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 | |||
2021-01-06 | Add CUDA support | Jordan Gong | |
2021-01-06 | Add TensorBoard support | Jordan Gong | |
2021-01-05 | Implement Batch All Triplet Loss | Jordan Gong | |
2021-01-03 | Unit testing on auto-encoder, HPM and Part Net | Jordan Gong | |
2020-12-29 | Encode class names to label and some access improvement | Jordan Gong | |
1. Encode class names using LabelEncoder from sklearn 2. Remove unneeded class variables 3. Protect some variables from being accessed in userspace | |||
2020-12-27 | Try some unit tests on CASIA-B dataset | Jordan Gong | |