#!/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

## Mean Search
# python src/evaluate_checkpoint.py \
#     -w TheThinker/ARC/sweet-forest-3149--checkpoint:v1 \
#     -g PATTERN \
#     -i mean \
#     --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 5096 \
#     --use-wandb True \
#     --multi-generate-per-pairs False \

# python src/evaluate_checkpoint.py \
#     -w TheThinker/ARC/sweet-forest-3149--checkpoint:v1 \
#     -g PATTERN \
#     -i gradient_ascent \
#     --num-steps 1 \
#     --lr 0.1 \
#     --lr-schedule true \
#     --optimizer adam \
#     --optimizer-kwargs '{"b2": 0.9}' \
#     --dataset-batch-size 32 \
#     --generator_n_pairs 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 \
#     --dataset-length 5096 \
#     --use-wandb True \
#     --multi-generate-per-pairs False \
#     --dataset-seed 3 \
#     --random-search-seed 3 \

# python src/evaluate_checkpoint.py \
#     -w TheThinker/ARC/sweet-forest-3149--checkpoint:v1 \
#     -g PATTERN \
#     -i gradient_ascent \
#     --num-steps 10 \
#     --lr 0.1 \
#     --lr-schedule true \
#     --optimizer adam \
#     --optimizer-kwargs '{"b2": 0.9}' \
#     --dataset-batch-size 32 \
#     --generator_n_pairs 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 \
#     --dataset-length 5096 \
#     --use-wandb True \
#     --multi-generate-per-pairs False \
#     --dataset-seed 3 \
#     --random-search-seed 3 \

# python src/evaluate_checkpoint.py \
#     -w TheThinker/ARC/sweet-forest-3149--checkpoint:v1 \
#     -g PATTERN \
#     -i gradient_ascent \
#     --num-steps 100 \
#     --lr 0.1 \
#     --lr-schedule true \
#     --optimizer adam \
#     --optimizer-kwargs '{"b2": 0.9}' \
#     --dataset-batch-size 32 \
#     --generator_n_pairs 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 \
#     --dataset-length 5096 \
#     --use-wandb True \
#     --multi-generate-per-pairs False \
#     --dataset-seed 3 \
#     --random-search-seed 3 \

python src/evaluate_checkpoint.py \
    -w TheThinker/ARC/sweet-forest-3149--checkpoint:v1 \
    -g PATTERN \
    -i gradient_ascent \
    --num-steps 400 \
    --lr 0.1 \
    --lr-schedule true \
    --optimizer adam \
    --optimizer-kwargs '{"b2": 0.9}' \
    --dataset-batch-size 32 \
    --generator_n_pairs 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 \
    --dataset-length 5096 \
    --use-wandb True \
    --multi-generate-per-pairs False \
    --dataset-seed 3 \
    --random-search-seed 3 