forked from uoguelph-mlrg/theano_alexnet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yaml
37 lines (29 loc) · 1012 Bytes
/
config.yaml
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
31
32
33
34
35
36
37
# If want to input None, use !!null
# Resume Training, start from scratch or resume training
resume_train: False
load_epoch: 20
# Learning Parameters
n_epochs: 60
learning_rate: 0.01 # def:0.01
lr_policy: step # def: step or auto
lr_adapt_threshold: 0.01
lr_step: [20, 40] # def: 20, learning rate will be reduced by 10 times at these steps
# Data
use_data_layer: False
rand_crop: True # def: True
shuffle: False # def: False, if shuffle the batches
para_load: True
batch_crop_mirror: False # if False, do randomly on each image separately
# Momentum
use_momentum : True # def: True
use_nesterov_momentum: False # def: False
momentum: 0.9 # def: 0.9
# Weight Decay
weight_decay: 0.0005 # def: 0.0005
# Directories
label_folder: /scratch/ilsvrc12/labels/
mean_file: /scratch/ilsvrc12/misc/img_mean.npy
weights_dir: ./models/ # directory for saving weights and results
snapshot_freq: 1 # frequency of saving weights
print_train_error: True
print_freq: 20 # frequency of printing training error rate