issues
search
ryanccarelli
/
ssljax
self-supervised learning in jax
GNU General Public License v2.0
8
stars
0
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Dev trainer refactor
#110
ryanccarelli
closed
2 years ago
0
add common loop utils
#109
ryanccarelli
closed
2 years ago
0
refactor optimizer for modules
#108
ryanccarelli
opened
2 years ago
0
Dev barlowtwins
#107
ryanccarelli
closed
2 years ago
0
vicreg config
#106
ryanccarelli
opened
2 years ago
1
vicreg loss
#105
ryanccarelli
opened
2 years ago
0
Dev data
#104
ryanccarelli
opened
2 years ago
0
Dev cleantask
#103
ryanccarelli
closed
2 years ago
0
add gradient clipping
#102
ryanccarelli
closed
2 years ago
0
Allow post process to be absent in the config
#101
AkashGanesan
closed
2 years ago
1
Add logging
#100
ryanccarelli
closed
2 years ago
0
Dev modules
#99
ryanccarelli
closed
2 years ago
0
simsiam loss and wsgd optimizer
#98
ryanccarelli
closed
2 years ago
0
Separate module from model definition, instantiate modules in SSLModel
#97
ryanccarelli
closed
2 years ago
0
clearly separate testing and baseline configs
#96
ryanccarelli
closed
2 years ago
0
integrate with scenic
#95
ryanccarelli
closed
2 years ago
0
Dev testing
#94
ryanccarelli
closed
2 years ago
2
SimSiam Shared Weights
#93
ryanccarelli
closed
2 years ago
0
add mocov3
#92
ryanccarelli
closed
2 years ago
0
Refactor Testing
#91
ryanccarelli
closed
2 years ago
0
updates to documentation
#90
ryanccarelli
closed
2 years ago
0
SWAV Augmentations/Pipeline
#89
ryanccarelli
opened
2 years ago
0
SWAV Prototype
#88
ryanccarelli
opened
2 years ago
0
SWAV Swapped Prediction Loss
#87
ryanccarelli
opened
2 years ago
0
SWAV Sinkhorn-Knopp Algorithm
#86
ryanccarelli
opened
2 years ago
0
b/16 vit configuration for imagenet
#85
ryanccarelli
closed
2 years ago
0
DINO VIT configuration
#84
ryanccarelli
closed
2 years ago
0
Improve schedulers
#83
ryanccarelli
closed
2 years ago
0
Setup eval loop and deprecate TorchData
#82
aranku
opened
2 years ago
1
EMA tau scheduler
#81
ryanccarelli
closed
2 years ago
0
Create BYOL ema scheduler
#80
AkashGanesan
closed
2 years ago
1
implement dino mlp and head, clean mlp implementation
#79
ryanccarelli
closed
2 years ago
0
Move pre and post process into step
#78
AkashGanesan
closed
2 years ago
0
Make pipeline jittable
#77
aranku
closed
2 years ago
0
Generalising Augmentation Distributions
#76
aranku
opened
2 years ago
0
Distributed sampler
#75
ryanccarelli
closed
2 years ago
0
Should we support batch norm with gradient accumulation
#74
AkashGanesan
closed
2 years ago
0
DINO documentation
#73
ryanccarelli
opened
2 years ago
0
DINO head
#72
ryanccarelli
closed
2 years ago
0
DINO centering and sharpening
#71
ryanccarelli
opened
2 years ago
0
Evaluations
#70
ryanccarelli
opened
2 years ago
0
Random crop augmentation is not random
#69
ryanccarelli
closed
2 years ago
0
DINO multicrop preprocessing
#68
ryanccarelli
closed
2 years ago
2
update readme
#67
ryanccarelli
closed
2 years ago
0
Fix bug in trainer that made it run slow
#66
aranku
closed
2 years ago
0
Dev dino
#65
ryanccarelli
closed
2 years ago
0
Cleanup how stateful params are being handled
#64
AkashGanesan
closed
2 years ago
0
Initialize branches and modules from pretrained
#63
ryanccarelli
closed
2 years ago
0
install readthedocs dependencies in workflow
#62
ryanccarelli
closed
2 years ago
0
Dev dino
#61
ryanccarelli
closed
2 years ago
0
Next