rename linear2 to MLPHead. Pretrain now gives a results csv (hopefully, untested)
Showing
- entcl/models/linear_head.py 1 addition, 1 deletionentcl/models/linear_head.py
- entcl/pretrain.py 21 additions, 27 deletionsentcl/pretrain.py
- entcl/run.py 9 additions, 6 deletionsentcl/run.py
- entcl/utils/findk.py 102 additions, 0 deletionsentcl/utils/findk.py
- entcl/utils/ncd.py 82 additions, 108 deletionsentcl/utils/ncd.py
- entcl/utils/ood.py 2 additions, 2 deletionsentcl/utils/ood.py
- entcl/utils/util.py 25 additions, 1 deletionentcl/utils/util.py
Loading
Please register or sign in to comment