summaryrefslogtreecommitdiff
path: root/models/model.py
AgeCommit message (Expand)Author
2021-02-20Merge branch 'master' into data_parallelJordan Gong
2021-02-20Separate triplet loss from modelJordan Gong
2021-02-19Correct cross reconstruction loss calculated in DataParallelJordan Gong
2021-02-19Merge branch 'master' into data_parallelJordan Gong
2021-02-19Allow evaluate unfinished modelJordan Gong
2021-02-18Merge branch 'master' into data_parallelJordan Gong
2021-02-18Implement adjustable input size and change some default configsJordan Gong
2021-02-18Decode mean appearance featureJordan Gong
2021-02-16Merge branch 'master' into data_parallelJordan Gong
2021-02-16Split transform and evaluate methodJordan Gong
2021-02-15Add DataParallel support on new codebaseJordan 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-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-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-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-05Change and improve weight initializationJordan Gong
2021-01-03Separate last fc matrix from weight init functionJordan Gong
2021-01-03Implement weight initializationJordan Gong
2021-01-03Update hyperparameter configuration, implement prototype fit functionJordan Gong
2020-12-29Correct batch splitterJordan Gong
2020-12-27Implement some parts of main model structureJordan Gong
2020-12-27Adopt type hinting generics in standard collections (PEP 585)Jordan Gong
2020-12-26Implement batch splitter to split sampled dataJordan Gong