summaryrefslogtreecommitdiff
path: root/models/model.py
AgeCommit message (Collapse)Author
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
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-03Implement weight initializationJordan Gong
2021-01-03Update hyperparameter configuration, implement prototype fit functionJordan Gong
2020-12-29Correct batch splitterJordan Gong
We can disentangle features from different subjects, but cannot do it at different temporal orders
2020-12-27Implement some parts of main model structureJordan Gong
1. Configuration parsers 2. Model signature generator
2020-12-27Adopt type hinting generics in standard collections (PEP 585)Jordan Gong
2020-12-26Implement batch splitter to split sampled dataJordan Gong
Disentanglement cannot be processed on different subjects at the same time, we need to load `pr` subjects one by one. The batch splitter will return a pr-length list of tuples (with 2 dicts containing k-length lists of labels, conditions, view and k-length tensor of clip data, representing condition 1 and condition 2 respectively).