#!/bin/sh

. ~/.bashrc

SCRATCH=${SCRATCH:"./"} # set somewhere with plenty of disk space
# HF_HOME=${SCRATCH:"./"} # optionally set HF_HOME to the same scratch space

. ~/pt/bin/activate # or elsehow activate python environment

model() {
    python -m lib.model -mc $SCRATCH $@
}

collect() {
    python collect.py -mc $SCRATCH -ds $DATASET_DIR $@
}

means() {
    collect -bs $3 -dev cuda:$2 -ms $1
}

covs() {
    collect -bs $3 -st covs -dev cuda:$2 -ms $1
}

analyze() {
    STATS_DIR=$SCRATCH/stats
    python analyze.py -mc $SCRATCH/TS -i $STATS_DIR/*/* -dev cuda:0 -o lc $@
}

progress() {
    analyze $@ -prog
}
