summaryrefslogtreecommitdiff
path: root/utils/configuration.py
AgeCommit message (Collapse)Author
2021-02-08Code refactoring, modifications and new featuresJordan Gong
1. Decode features outside of auto-encoder 2. Turn off HPM 1x1 conv by default 3. Change canonical feature map size from `feature_channels * 8 x 4 x 2` to `feature_channels * 2 x 16 x 8` 4. Use mean of canonical embeddings instead of mean of static features 5. Calculate static and dynamic loss separately 6. Calculate mean of parts in triplet loss instead of sum of parts 7. Add switch to log disentangled images 8. Change default configuration
2021-01-23Add late start support for non-disentangling partsJordan Gong
2021-01-14Enable optimizer fine tuningJordan Gong
2021-01-13Add multiple checkpoints for different model and set default config valueJordan Gong
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-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-03Update hyperparameter configuration, implement prototype fit functionJordan Gong
2020-12-29Encode class names to label and some access improvementJordan 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-27Make naming scheme consistentJordan Gong
Use `dir` instead of `path`
2020-12-27Add dataset selector to config type hint, change ClipLabels typo to ClipViewsJordan Gong
2020-12-27Adopt type hinting generics in standard collections (PEP 585)Jordan Gong
2020-12-26Add config file and corresponding type hintJordan Gong