Skip to content

Commit

Permalink
Add resnet 4/4 script
Browse files Browse the repository at this point in the history
  • Loading branch information
gzuidhof committed Jun 21, 2016
1 parent d3acbd7 commit c9ca7d1
Show file tree
Hide file tree
Showing 3 changed files with 76 additions and 2 deletions.
4 changes: 2 additions & 2 deletions config/default_resnet.ini
Original file line number Diff line number Diff line change
Expand Up @@ -27,8 +27,8 @@ optimization: -
learning_rate: 0 ;not used, see schedule
momentum: 0.92
l2_lambda: 0.00005
batch_size_train: 36
batch_size_validation: 80
batch_size_train: 12
batch_size_validation: 36
n_epochs: 200

[preprocessing] ;Not used
Expand Down
53 changes: 53 additions & 0 deletions config/resnet44.ini
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
[info]
experiment: ResNet default settings
name: resnet44_89

[dataset]
pixels: 96
n_classes: 2
channels: 1
subset: None
filenames_train: /scratch-shared/vdgugten/data/cadOWN_0.5mm_96x96_xy_xz_yz/subset[0123456]/*/*.pkl.gz
filenames_validation: /scratch-shared/vdgugten/data/cadOWN_0.5mm_96x06_xy_xz_yz/subset[7]/*/*.pkl.gz
data_folder: None ;not used

[network]
architecture: resnet
input_size: 64
depth: 4
branching_factor: 4
batch_normalization: True
batch_normalization_alpha: 0.1
dropout: 0.25
spatial_dropout: 0.5
gaussian_noise: 0.05

[updates]
optimization: -
learning_rate: 0 ;not used, see schedule
momentum: 0.92
l2_lambda: 0.00005
batch_size_train: 128
batch_size_validation: 256
n_epochs: 200

[preprocessing] ;Not used
random_crop: 0 ;no cropping
erode_segmentation: 11

[normalization]
zero_center: True
mean_pixel: 0.2606236106512

[augmentation]
augment: True
flip: True
zoom: 0.1
rotation: 20
translation: 3


[misc]
multiprocess_load_augmentation: False
save_every_n_epoch: 2
n_workers_load_augmentation: 10
21 changes: 21 additions & 0 deletions scripts/resnet_scripts/resnet44_01.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
#!/bin/bash
#SBATCH -t 2-00:00:00
#SBATCH -p gpu
#SBATCH -c 12

#Prepare python environment
export PYTHONPATH=$HOME/pythonpackages/lib/python:$PYTHONPATH
export PYTHONPATH=$HOME/pythonpackages/lib/python2.7/site-packages:$PYTHONPATH
module load python/2.7.9
module load cuda
module load cudnn

#Go to project folder
cd $HOME/luna16/src/deep


#Go!!!

echo "starting python"
export THEANO_FLAGS='mode=FAST_RUN,device=gpu,floatX=float32,lib.cnmem=1'
srun -u python train.py ../../config/default_fr3dnet.ini

0 comments on commit c9ca7d1

Please sign in to comment.