James Betker
|
79367f753d
|
Fix error & add nonfinite warning
|
2021-11-09 23:58:41 -07:00 |
|
James Betker
|
438217094c
|
Also debug distribution of switch
|
2021-06-07 15:36:07 -06:00 |
|
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 |
|
James Betker
|
119f17c808
|
Add testing capabilities for segformer & contrastive feature
|
2021-04-27 09:59:50 -06:00 |
|
James Betker
|
9bbe6fc81e
|
Get segformer to a trainable state
|
2021-04-25 11:45:20 -06:00 |
|
James Betker
|
fc623d4b5a
|
Add segformer model. Start work on BYOL adaptation that will support training it.
|
2021-04-23 17:16:46 -06:00 |
|