* add necks * refactor linear neck * rename simmim neck * add heads * add losses * fix * add unittest * update * update cae * remove mim head * update config |
||
---|---|---|
.. | ||
__init__.py | ||
asymmetric_loss.py | ||
cae_loss.py | ||
cosine_similarity_loss.py | ||
cross_correlation_loss.py | ||
cross_entropy_loss.py | ||
focal_loss.py | ||
label_smooth_loss.py | ||
reconstruction_loss.py | ||
seesaw_loss.py | ||
swav_loss.py | ||
utils.py |