da1da48eb6
* Add channel argments to mae_head When trying iTPN pretrain, it only supports images with 3 channels. One of the restrictions is from MAEHead. * Transfer other argments from iTPNHiViT to HiViT The HiViT supports specifying channels, but the iTPNHiViT class can't pass channel argments to it. This is one of the reasons that iTPNHiViT implementation only support images with 3 channels. * Update itpn.py Fix hint problem |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| barlowtwins.py | ||
| base.py | ||
| beit.py | ||
| byol.py | ||
| cae.py | ||
| densecl.py | ||
| eva.py | ||
| itpn.py | ||
| mae.py | ||
| maskfeat.py | ||
| mff.py | ||
| milan.py | ||
| mixmim.py | ||
| moco.py | ||
| mocov3.py | ||
| simclr.py | ||
| simmim.py | ||
| simsiam.py | ||
| spark.py | ||
| swav.py | ||