-
Notifications
You must be signed in to change notification settings - Fork 0
/
slurm_script.sh
30 lines (23 loc) · 1.08 KB
/
slurm_script.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#!/bin/bash
#SBATCH --job-name=pointnet
#SBATCH --nodes=1
#SBATCH --gres=gpu:1 # Request one GPU
#SBATCH --cpus-per-task=8 # Number of CPU cores per task
#SBATCH --time=10:00:00 # Request 1 hour of runtime
#SBATCH --mem=16GB # Request 16GB of RAM
#SBATCH --partition=v100 # Specify the GPU partition
#SBATCH -o slurm_outputs/slurm-output-%j.txt # Write the output to specific directory
# Load required modules (if necessary)
# module load cuda/11.0 cudnn/8.0.4
# module load anaconda3/2020.11
# Activate your Python environment (if using a virtual environment)
# source activate myenv
# Activate your Conda environment (if using a Conda environment)
# conda activate py311
# # Run your PyTorch script
# python --version
# python train.py --model pointnet_cls --dataset ModelNet40 --epoch 100 --batch_size 32
# python train.py --model pointcnn_v1_cls --dataset ModelNet40_hdf5 --epoch 100 --batch_size 32
python train.py --model pointnet_cls --dataset TUBerlin --epoch 500 --batch_size 32 --num_category 250
# Jupyter Environment
# jupyter notebook --ip=0.0.0.0 --port=3001 --no-browser