-
Notifications
You must be signed in to change notification settings - Fork 2
/
job.sh
21 lines (17 loc) · 754 Bytes
/
job.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#!/bin/bash
#SBATCH --job-name=train-pc-cls
#SBATCH --partition=gpu
#SBATCH --nodes=1 # number of nodes
#SBATCH --mem=64GB # memory per node in MB (different units with$
#SBATCH --ntasks-per-node=16 # number of cores
#SBATCH --time=24:00:00 # HH-MM-SS
#SBATCH --output log/job_tf.%N.%j.out # filename for STDOUT (%N: nodename, %j: j$
#SBATCH --error log/job_tf.%N.%j.err # filename for STDERR
# load cuda
module load cuda/9.2
# activate environment
source ~/anaconda3/bin/activate ~/anaconda3/envs/pc-classification
# locate to your root directory
cd /home/mwasil2s/deeplearning_walkthrough/pointcloud_classification
# run the script
python trainer.py --model DGCNNC --train