#!/bin/bash

ENV_NAME=mvq

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

CONFIG=64-MVQGAN-IN.yaml
DATASET=imagenet64
MODE=train
STAGE=vae

cd ..

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