aboutsummaryrefslogtreecommitdiff
path: root/posrecon
AgeCommit message (Collapse)Author
2022-08-27Make InfoNCE loss distributableHEADmasterJordan Gong
2022-08-26Rewrite SimCLR-based PosRecon model with some tweaksJordan Gong
- No inheritance, encapsulation and immutability, making code more readable - Remove mask token and encode visible patches only, inspired from MAE and MSN - Switch from batch-level randomization (shuffling and masking) to sample-level, and remove shuffle ratio for simplicity - Compute loss during forward
2022-08-25Add a dirty version of PosRecon trainerJordan Gong
2022-08-21Some modifications for PosRecon trainerJordan Gong
2022-08-19Add ViT with position shuffling/unshuffling and masking methodsJordan Gong
2022-08-19Add dependenciesJordan Gong
- Pytorch Lighting (pytorch-lighting) - Pytorch image models (timm)
2022-08-19Add ViT with position shuffling/unshuffling methodsJordan Gong