#!/bin/bash

NUM_PROC=$(wc -w <<< $(tr ',' ' ' <<< $CUDA_VISIBLE_DEVICES))
MASTER_PORT="<master port>" 
BATCH_SIZE_PER_GPU="<batch size per 1 gpu>"

DATA_DIR="<data dir>"
MODEL="<model>"
EXP="<experiment name>"

# if using W&B for logging
export WANDB_ENTITY="<W&B user name>"
export WANDB_PROJECT="<project name>"
export WANDB_NAME="<run name>"

python one_shot_pruning.py \
    \
    --data-dir ${DATA_DIR} \
    \
    --sparseml-recipe "<sparseml recipe path>" \
    \
    --model ${MODEL} \
    \
    --experiment ${EXP} \
    \
    -b ${OBS_BATCH_SIZE} \
    \
    --sparsities "<list of sparsitites>" 
