Adds cleanup_data.sh

cvxgrp · Dec 2, 2024 · 47f3db3 · 47f3db3
1 parent 09df0df
commit 47f3db3
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/utils/cleanup_data.sh b/utils/cleanup_data.sh
@@ -0,0 +1,20 @@
+#!/bin/bash
+#SBATCH --job-name=adelie_sweep
+#SBATCH --output=adelie_sweep/output/slurm-%A_%a.out
+#SBATCH --ntasks=1
+#SBATCH --cpus-per-task=1
+#SBATCH --mem-per-cpu=64GB
+#SBATCH --partition=candes
+
+BASE_DIR=$HOME/adelie_alo/benchmarking/lasso_sweep
+RESULTS_DIR=$BASE_DIR/results
+mkdir -p $RESULTS_DIR
+DEST_FILE=$RESULTS_DIR/sweep.npz
+
+
+ml python/3.12.1
+ml py-pytorch/2.4.1_py312
+. $HOME/randalo/.venv/bin/activate
+
+
+python $HOME/randalo/utils/cleanup_adelie_data.py