VIT-VQGAN

Run code on TPU VMs

Once you've created a TPU VM, run

python tpu_run.py

Basically this SSH's into your TPU VM, installs dependencies, and then runs your command. You should modify the code in tpu_run.py and configuration in pretrain/configs/*.yaml.

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
additional_file		additional_file
configs		configs
data		data
examples		examples
models		models
training		training
README.md		README.md
audio_diff.py		audio_diff.py
config.py		config.py
dataset_test.py		dataset_test.py
network_test.py		network_test.py
run.sh		run.sh
setup.py		setup.py
tpu_run.py		tpu_run.py
tpu_startup_script.sh		tpu_startup_script.sh
train.py		train.py

jiasenlu/vit-vqgan-jax

Folders and files

Latest commit

History

Repository files navigation

VIT-VQGAN

Run code on TPU VMs

About

Resources

Stars

Watchers

Forks

Languages