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
|
1a2b9fa130
|
Get rid of old byol net wrapping
Simplifies and makes this usable with DLAS' multi-gpu trainer
|
2021-04-27 12:48:34 -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 |
|