mvq / bin / train_cifar10.sh
train_cifar10.sh
Raw
#!/bin/bash

ENV_NAME=mvq

source $HOME/envs/$ENV_NAME/bin/activate

CONFIG=32-MVQGAN.yaml
DATASET=cifar10
MODE=train
STAGE=vae

cd ..

torchrun --standalone main.py -c $CONFIG -m $MODE -s $STAGE -d $DATASET