train_vae_simple

#!/bin/bash

DATASET="$1"
LR="0.0005"
W1="-0.25"
INNER="PSICNN"

if [ -z "$DATASET" ]; then
    echo "NO DATASET"
    exit
fi

OUTDIR="experiments/vae_simple/${DATASET}"
mkdir -p "$OUTDIR"
MODEL="vae_trajectory[w=$W1,projfn=$INNER]"

date >> "$OUTDIR/progress.txt"
./.colorize ./train.py                      \
    --log-to "runs/$OUTDIR"                 \
    --batch-size 300                        \
    --learning-rate "$LR"                   \
    --epochs 150                            \
    imagepairs[files=youtube/$DATASET/*]    \
    "$MODEL"                                \
    "$OUTDIR/checkpoint_{epoch:0>5}.pth" | tee -a "$OUTDIR/progress.txt"

./render_vae "$DATASET" "$MODEL" "$OUTDIR"