Datasetpublic

MTDoven/ViTTiny1022

ViTTiny1022 The dataset for Scaling Up Parameter Generation: A Recurrent Diffusion Approach. Requirement Install torch and other dependencies conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-cuda=12.1 -c pytorch -c nvidia pip install timm einops seaborn openpyxl Usage Test one checkpoint cd ViTTiny1022 python test.py ./chechpoint_test/0000_acc0.9613_class0314_condition_cifar10_vittiny.pth #… See the full description on the dataset page: https://huggingface.co/datasets/MTDoven/ViTTiny1022.

sourceHugging Faceunknownupdated 2y agoView on Hugging Face

2likes2.8kdownloads

Dataset Card

ViTTiny1022

The dataset for Scaling Up Parameter Generation: A Recurrent Diffusion Approach.

Requirement

Install torch and other dependencies

shell

conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-cuda=12.1 -c pytorch -c nvidia
pip install timm einops seaborn openpyxl

Usage

Test one checkpoint

shell

cd ViTTiny1022
python test.py ./chechpoint_test/0000_acc0.9613_class0314_condition_cifar10_vittiny.pth
# python test.py <path/to/the/file/you/want/to/test.pth>

Test some checkpoints

shell

cd ViTTiny1022
python test.py ./chechpoint_test
# python test.py <path/to/the/folder/containing/files/you/want/to/test>

Test the detail of one checkpoint

shell

cd ViTTiny1022
python detail.py ./chechpoint_test/0000_acc0.9613_class0314_condition_cifar10_vittiny.pth
# python detail.py <path/to/the/file/you/want/to/test.pth>

Train from source

shell

cd ViTTiny1022
bash train.sh
bash split.sh