#!/bin/bash -i
#SBATCH --mem=60g
#SBATCH --cpus-per-task=16
#SBATCH --time=14-0
#SBATCH --gres=gpu:1
#SBATCH --mail-user=EMAIL
#SBATCH --mail-type=ALL

cd PATH
source PATH/bin/activate
python -m scripts.fidelity.isc.distribution \
  --folder PATH/datasets/imagenet-64x64/images \
  --image_height 64 \
  --image_width 64 \
  --image_channels 3 \
  --num_samples 1281167 \
  --batch_size 100 \
  --distribution_save_path results_fidelity_isc/imagenet-64x64/distribution.npy \
  --classes_save_path results_fidelity_isc/imagenet-64x64/classes.npy
