Supported Models

Category	Task	Name	Sub-style
CNN	Classification	ResNet18
CNN	Classification	ResNet50
CNN	Classification	MobileNet	mobilenetv3_small
CNN	Classification	MobileNet	mobilenetv3_large
CNN	Classification	EfficientNet	efficientnet_v2_s
CNN	Classification	EfficientNet	efficientnet_v2_m
CNN	Classification	EfficientNet	efficientnet_v2_l
ViT	Classification	PVT-V1	pvt_tiny
ViT	Classification	PVT-V1	pvt_small
ViT	Classification	PVT-V1	pvt_medium
ViT	Classification	PVT-V1	pvt_large
ViT	Classification	PVT-V2	pvt_v2_b0
ViT	Classification	PVT-V2	pvt_v2_b1
ViT	Classification	PVT-V2	pvt_v2_b2
ViT	Classification	PVT-V2	pvt_v2_b3
ViT	Classification	PVT-V2	pvt_v2_b4
ViT	Classification	PVT-V2	pvt_v2_b5
ViT	Classification	DeiT	deit_tiny_224
ViT	Classification	DeiT	deit_small_224
ViT	Classification	DeiT	deit_base_224
ViT	Classification	CSwin	cswin_64_tiny
ViT	Classification	CSwin	cswin_64_small
ViT	Classification	CSwin	cswin_96_base
ViT	Classification	CSwin	cswin_144_large
Transformers	LM	BERT	bert-base-uncased
Transformers	LM	GPT2	gpt2
Transformers	Classification	RoBERTa	roberta-base
Transformers	Classification	RoBERTa	roberta-large
Transformers	LM	OPT	facebook/opt-125m
Transformers	LM	OPT	facebook/opt-350m
Transformers	LM	OPT	facebook/opt-1.3b
Transformers	LM	OPT	facebook/opt-2.7b
Transformers	LM	OPT	facebook/opt-13b
Transformers	LM	OPT	facebook/opt-30b
Transformers	LM	OPT	facebook/opt-66b
Transformers	LM	GPT-NEO	EleutherAI/gpt-neo-125M
Transformers	LM	GPT-NEO	EleutherAI/gpt-neo-1.3B
Transformers	LM	GPT-NEO	EleutherAI/gpt-neo-2.7B
Transformers	LM	GPT-NEO	EleutherAI/gpt-neox-20b
Transformers	Translation	T5	t5-small
Transformers	Translation	T5	t5-base
Transformers	Translation	T5	t5-large
Transformers	Translation	T5	google/t5-v1_1-small