scripts/flywire/ablations/max_neurons_merged.sh
#!/bin/bash

max_neurons_merged=(1 2 3 4 5 6 7)

for n in "${max_neurons_merged[@]}"
do
    bsub -J "AE Eval (Topology)" -n 8 -gpu "num=1" -q gpu_h100 -o "logs/paper_flywire_n_neurons_ablation_max_${n}.log" python eval_topology.py \
        --model ae_d1024_m512 \
        --pth /nrs/turaga/jakob/implicit-neurons/ckpt/ae/affinity_benchmark_w_dust_frag_v2/ckpt/checkpoint-946.pth \
        --data_path /nrs/turaga/jakob/implicit-neurons/flywire_full_v783/train \
        --neuron_id_path /nrs/turaga/jakob/implicit-neurons/flywire_full_v783/affinity/opticlobe_family/affinity_test_paper.csv \
        --output_dir "/nrs/turaga/jakob/implicit-neurons/ckpt/ae/paper_flywire_n_neurons_ablation_max_${n}" \
        --fam_to_id_mapping /nrs/turaga/jakob/implicit-neurons/flywire_full_v783/types/visual_neurons_family_to_id.json \
        --num_workers 6 \
        --point_cloud_size 1024 \
        --batch_size 1 \
        --data_global_scale_factor 659.88367 \
        --thresholds 0.8 \
        --max_neurons_merged "${n}"
done