summaryrefslogtreecommitdiff
path: root/models
AgeCommit message (Expand)Author
2021-02-27Implement Batch Hard triplet loss and soft marginJordan Gong
2021-02-26Fix predict functionJordan Gong
2021-02-21Remove FConv blocksJordan Gong
2021-02-20Separate triplet loss from modelJordan Gong
2021-02-19Allow evaluate unfinished modelJordan Gong
2021-02-18Implement adjustable input size and change some default configsJordan Gong
2021-02-18Remove 1x1 conv layers when not usedJordan Gong
2021-02-18Decode mean appearance featureJordan Gong
2021-02-18Decode mean appearance featureJordan Gong
2021-02-16Split transform and evaluate methodJordan Gong
2021-02-15Revert "Memory usage improvement"Jordan Gong
2021-02-14Memory usage improvementJordan Gong
2021-02-14Prepare for DataParallelJordan Gong
2021-02-10Save scheduler state_dictJordan Gong
2021-02-09Improve performance when disentanglingJordan Gong
2021-02-09Some optimizationsJordan Gong
2021-02-08Code refactoring, modifications and new featuresJordan Gong
2021-01-23Remove the third term in canonical consistency lossJordan Gong
2021-01-23Add late start support for non-disentangling partsJordan Gong
2021-01-23Transform all frames together in evaluationJordan Gong
2021-01-23Evaluation bug fixes and code reviewJordan Gong
2021-01-22Handle unexpected restore iterJordan Gong
2021-01-21Print average losses after 100 itersJordan Gong
2021-01-21Bug fixesJordan Gong
2021-01-14Enable optimizer fine tuningJordan Gong
2021-01-14Remove DataParallelJordan Gong
2021-01-13Update config file and convert int to str when joiningJordan Gong
2021-01-13Add multiple checkpoints for different model and set default config valueJordan Gong
2021-01-12Move the model to GPU before constructing optimizerJordan Gong
2021-01-12Some changes in hyperparameter configJordan Gong
2021-01-12Some type hint fixesJordan Gong
2021-01-12Typo correct in evaluate functionJordan Gong
2021-01-11Add evaluation script, code review and fix some bugsJordan Gong
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-07Train different models in different conditionsJordan Gong
2021-01-07Add typical training script and some bug fixesJordan Gong
2021-01-07Change device config and add enable multi-GPU computingJordan Gong
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
2021-01-03Separate last fc matrix from weight init functionJordan Gong
2021-01-03Delete dead training judgeJordan Gong
2021-01-03Implement weight initializationJordan Gong
2021-01-03Update hyperparameter configuration, implement prototype fit functionJordan Gong
2021-01-03Add separate fully connected layersJordan Gong