summaryrefslogtreecommitdiff
path: root/models/model.py
AgeCommit message (Collapse)Author
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).