summaryrefslogtreecommitdiff
AgeCommit message (Collapse)Author
2021-01-22Handle unexpected restore iterJordan Gong
1. Skip finished model before load it 2. Raise error when restore iter is greater than total iter
2021-01-21A type hint fixJordan Gong
2021-01-21Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # utils/configuration.py
2021-01-21Print average losses after 100 itersJordan Gong
2021-01-21Bug fixesJordan Gong
1. Turn off autograd while decoding canonical and pose features 2. Change default batch size to (4, 8)
2021-01-20Bump up version for scikit-learnJordan Gong
2021-01-14Enable optimizer fine tuningJordan Gong
2021-01-14Remove DataParallelJordan Gong
2021-01-14Remove DataParallelJordan Gong
2021-01-13Remove more stuffJordan Gong
2021-01-13Tensorboard typo fixJordan Gong
2021-01-13Add server startup scriptJordan Gong
2021-01-13Merge branch 'master' into python3.8Jordan Gong
2021-01-13Update config file and convert int to str when joiningJordan Gong
2021-01-13Type hint fixesJordan Gong
2021-01-13Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/model.py
2021-01-13Add multiple checkpoints for different model and set default config valueJordan Gong
2021-01-12Merge branch 'master' into python3.8Jordan Gong
2021-01-12Move the model to GPU before constructing optimizerJordan Gong
2021-01-12Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # utils/configuration.py
2021-01-12Some changes in hyperparameter configJordan Gong
1. Separate hyperparameter configs in model, optimizer and scheduler 2. Add more tunable hyperparameters in optimizer and scheduler
2021-01-12Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/model.py # utils/dataset.py
2021-01-12Some type hint fixesJordan Gong
2021-01-12Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/model.py
2021-01-12Print result after evaluationJordan Gong
2021-01-12Typo correct in evaluate functionJordan Gong
2021-01-12Bump up version for tqdmJordan Gong
2021-01-11Add evaluation script, code review and fix some bugsJordan 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-11Implement evaluatorJordan Gong
2021-01-10Make predict function transform samples different conditions in a single shotJordan Gong
2021-01-09Add prototype predict functionJordan Gong
2021-01-09Change auto-encoder input in evaluationJordan Gong
2021-01-09Fix NaN when separate sum is zeroJordan Gong
2021-01-07Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/model.py
2021-01-07Train different models in different conditionsJordan Gong
2021-01-07Type hint for python version lower than 3.9Jordan Gong
2021-01-07Merge remote-tracking branch 'origin/python3.8' into python3.8Jordan Gong
2021-01-07Type hint for python version lower than 3.9Jordan Gong
2021-01-07Type hint for python version lower than 3.9Jordan Gong
2021-01-07Type hint for python version lower than 3.9Jordan 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
2021-01-07Change device config and add enable multi-GPU computingJordan Gong
1. Add `disable_acc` switch for disabling accelerator. When it is off, system will automatically choosing accelerator. 2. Enable multi-GPU training using torch.nn.DataParallel
2021-01-06Add CUDA supportJordan Gong
2021-01-06Add TensorBoard supportJordan Gong
2021-01-05Implement checkpoint mechanismJordan Gong
2021-01-05Implement Batch All Triplet LossJordan Gong
2021-01-05Change and improve weight initializationJordan Gong
1. Change initial weights for Conv layers 2. Find a way to init last fc in init_weights
2021-01-03Separate last fc matrix from weight init functionJordan Gong
Recursive apply will override other parameters too
2021-01-03Delete dead training judgeJordan Gong
2021-01-03Implement weight initializationJordan Gong