summaryrefslogtreecommitdiff
path: root/utils
AgeCommit message (Collapse)Author
2021-01-09Fix NaN when separate sum is zeroJordan 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-05Implement Batch All Triplet LossJordan Gong
2021-01-03Update hyperparameter configuration, implement prototype fit functionJordan Gong
2020-12-31Make super class constructor revoke consistentJordan Gong
2020-12-29Add type hint for new label (numpy.int64)Jordan 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-27Prepare for FVG datasetJordan Gong
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-26Sample k more clips for disentanglementJordan Gong
2020-12-26Add config file and corresponding type hintJordan Gong
2020-12-26Combine transformed height and width to `frame_size`Jordan Gong
2020-12-21Change image loading techniqueJordan Gong
1. Use Pillow.Image.open instead of torchvision.io.read_image to read image 2. Transforming PIL images instead of tensors which performs better and device option is removed 3. Images are normalized now
2020-12-191. Delete unused transform functionJordan Gong
2. Reorganize the initialization cache dicts
2020-12-19Fix indexing error when no clip to be discardJordan Gong
2020-12-19Add cache switch, allowing load all data into RAM before samplingJordan Gong
2020-12-18Implement triplet samplerJordan Gong
2020-12-18Implement CASIA-B datasetJordan Gong