Skip to content

Commit dfbb116

Browse files
committed
chore: cleanup scripts
1 parent a1db014 commit dfbb116

3 files changed

Lines changed: 39 additions & 246 deletions

File tree

scripts/train_all_models.sh

Lines changed: 0 additions & 125 deletions
This file was deleted.
Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,39 @@
1+
#!/bin/bash
2+
# Distributed training script for license classifier (copyright + license models)
3+
# Uses GRU architecture with multi-GPU support
4+
#
5+
# Generated for DeLFT
6+
7+
set -e # Exit on error
8+
9+
# Common SLURM configuration for multi-GPU training
10+
SRUN_OPTS="--container-mounts=/netscratch:/netscratch,$HOME:$HOME \
11+
--container-workdir=/netscratch/lfoppiano/delft \
12+
--container-image=/netscratch/lfoppiano/enroot/tensorflow-2.17.2-gpu-delft-updated.sqsh \
13+
--mem=200G \
14+
-p V100-32GB,RTX3090,RTXA6000 \
15+
--gpus=4 \
16+
--nodes=1 \
17+
--time=3-00:00"
18+
19+
PYTHON_CMD=".venv/bin/python -m delft.applications.licenseClassifier"
20+
21+
# Training parameters
22+
ARCHITECTURE="gru"
23+
FOLD_COUNT=4
24+
NUM_WORKERS=1
25+
26+
echo "=========================================="
27+
echo "Distributed training - License classifier"
28+
echo "Architecture: ${ARCHITECTURE}"
29+
echo "Folds: ${FOLD_COUNT}"
30+
echo "GPUs: 4"
31+
echo "=========================================="
32+
33+
srun $SRUN_OPTS $PYTHON_CMD train \
34+
--architecture ${ARCHITECTURE} \
35+
--num-workers ${NUM_WORKERS}
36+
37+
echo "=========================================="
38+
echo "Training completed"
39+
echo "=========================================="

scripts/train_eval_all_models.sh

Lines changed: 0 additions & 121 deletions
This file was deleted.

0 commit comments

Comments
 (0)