#!/bin/bash

# Source the bash profile to load Conda properly
source ~/.bashrc

export PYTHONPATH=$PYTHONPATH:/home/mvm/arc

# Activate your Conda environment
conda activate ditto_env


# python src/evaluate_checkpoint.py \
#     -w TheThinker/ARC/restful-spaceship-3150--checkpoint:v1 \
#     -g PATTERN \
#     -i transductive \
#     --dataset-batch-size 128 \
#     --generator_n_pairs 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 \
#     --dataset-length 128 \
#     --use-wandb True \
#     --multi-generate-per-pairs False \

python src/evaluate_checkpoint.py \
    -w TheThinker/ARC/restful-spaceship-3150--checkpoint:v1 \
    -g PATTERN \
    -i ttt \
    --num-steps 10 \
    --lr 1e-3 \
    --dataset-batch-size 8 \
    --generator_n_pairs 3 10 50 100 200 400 500 \
    --dataset-length 128 \
    --use-wandb True \
    --multi-generate-per-pairs False
