diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-26 17:30:37 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-26 17:30:37 +0800 |
| commit | ea370e306eec3a5191bd838f7f650658e0fbb115 (patch) | |
| tree | 4eca04733b3d9b99b97cc2762c92480c5fa14a87 /libs/criteria.py | |
| parent | 287c2522c83b37718ddd4ad3772e687298a0bb8a (diff) | |
Rewrite SimCLR-based PosRecon model with some tweaks
- 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
Diffstat (limited to 'libs/criteria.py')
0 files changed, 0 insertions, 0 deletions