Age | Commit message (Collapse) | Author |
|
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
|
|
|
|
|
|
|
|
1. Change initial weights for Conv layers
2. Find a way to init last fc in init_weights
|
|
Recursive apply will override other parameters too
|
|
|
|
|
|
We can disentangle features from different subjects, but cannot do it at different temporal orders
|
|
1. Configuration parsers
2. Model signature generator
|
|
|
|
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).
|