Add pre-trained models and enhance training with learning rate scheduler
- Added pre-trained models for DINO and DINOv2 to backbone_store - Updated .gitignore to exclude egg-info files - Enhanced LinearHead to ensure new weights are initialized correctly - Integrated Cosine Annealing LR Scheduler into pretraining and continual learning sessions - Improved OOD accuracy computation with additional entropy and energy metrics
Showing
- .gitignore 3 additions, 0 deletions.gitignore
- entcl/cl.py 15 additions, 8 deletionsentcl/cl.py
- entcl/models/backbone_store/dino_vitbase16_pretrain.pth 3 additions, 0 deletionsentcl/models/backbone_store/dino_vitbase16_pretrain.pth
- entcl/models/backbone_store/dinov2_vitb14_pretrain.pth 3 additions, 0 deletionsentcl/models/backbone_store/dinov2_vitb14_pretrain.pth
- entcl/models/linear_head.py 3 additions, 0 deletionsentcl/models/linear_head.py
- entcl/models/model.py 54 additions, 10 deletionsentcl/models/model.py
- entcl/pretrain.py 15 additions, 0 deletionsentcl/pretrain.py
- entcl/run.py 10 additions, 8 deletionsentcl/run.py
- entcl/utils/ncd.py 41 additions, 32 deletionsentcl/utils/ncd.py
- entcl/utils/ood.py 11 additions, 8 deletionsentcl/utils/ood.py
- experiments/experiments3.ipynb 854 additions, 0 deletionsexperiments/experiments3.ipynb
Loading
Please register or sign in to comment