UniNet
UniNet copied to clipboard
Sense-X
Unified Architecture Search with Convolution, Transformer, and MLP (ECCV 2022)
Pytorch implementation of UniNet (ECCV 2022)
This repo is the offcial implementation of the paper UniNet: Unified Architecture Search with Convolution, Transformer, and MLP
@article{UniNet,
author = {Jihao Liu, Xin Huang, Guanglu Song, Yu Liu, Hongsheng Li},
journal = {arXiv:2207.05420},
title = {UniNet: Unified Architecture Search with Convolution, Transformer, and MLP},
year = {2022},
}
Update
20/12/2022 Update pretrained models.
25/10/2022 Update the source code.
Environment
The code is tested with torch==1.11 and timm==0.5.4.
Availble models
| Models | Params (M) | FLOPs (G) | Pretrain Epochs | Top-1 Acc. | ckpt |
|---|---|---|---|---|---|
| UniNet-B1 | 11.5 | 1.1 | 300 | 81.0 | ckpt |
Run experiments
Currently, we supporting running experiments with slurm. You can reproduce the results of UniNet-B1 as follows:
sh exp/b1/run.sh partition 8
Sense-X