Ross Wightman 5f47518f27 Fix pit implementation to be clsoer to deit/levit re distillation head handling

2022-03-21 11:12:14 -07:00

..

data

Significant model refactor and additions:

2022-02-28 13:56:23 -08:00

loss

…

models

Fix pit implementation to be clsoer to deit/levit re distillation head handling

2022-03-21 11:12:14 -07:00

optim

Fix bug introduced in non layer_decay weight_decay application. Remove debug print, fix arg desc.

2022-02-28 17:06:32 -08:00

scheduler

Significant model refactor and additions:

2022-02-28 13:56:23 -08:00

utils

…

__init__.py

…

version.py

…