James Betker f129eaa39e Clean up byol a bit ... - Remove option to aug in dataset (there's really no reason for this now that kornia works on GPU on windows) - Other stufff		2021-05-24 21:35:46 -06:00
..
__init__.py	Get segformer to a trainable state	2021-04-25 11:45:20 -06:00
backbone.py	Clean up byol a bit	2021-05-24 21:35:46 -06:00
segformer.py	Add testing capabilities for segformer & contrastive feature	2021-04-27 09:59:50 -06:00