Skip to content

Commit 88965b3

Browse files
committed
Update template-common/config.yaml
To resolve issue pytorch-ignite#293
1 parent e273a27 commit 88965b3

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

src/templates/template-common/config.yaml

+7-7
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
seed: 777
2-
data_path: ./
3-
batch_size: 32
4-
eval_batch_size: 32
5-
num_workers: 4
6-
max_epochs: 20
7-
use_amp: false
1+
seed: 777 # random seed
2+
data_path: ./ # input data path
3+
batch_size: 32 # train batch size
4+
eval_batch_size: 32 # evaluation batch size
5+
num_workers: 4 # number of subprocesses to use for data loading
6+
max_epochs: 20 # number of maximum epochs
7+
use_amp: false # use AMP (Automatic Mixed Precision)
88
debug: false
99
train_epoch_length: null
1010
eval_epoch_length: null

0 commit comments

Comments
 (0)