Skip to content

Commit 8d41c73

Browse files
Venkat Ramnan KalyanakumarVenkat Ramnan Kalyanakumar
Venkat Ramnan Kalyanakumar
authored and
Venkat Ramnan Kalyanakumar
committed
feat: config for matching ignite capapbility
1 parent 2d50595 commit 8d41c73

File tree

1 file changed

+19
-3
lines changed
  • src/templates/template-vision-classification

1 file changed

+19
-3
lines changed
Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,19 @@
1-
#::= from_template_common ::#
2-
lr: 0.0001 # learning rate
3-
model: resnet18 # model name
1+
seed: 777 # random seed
2+
data_path: ./ # input data path
3+
batch_size: 400 # train batch size
4+
eval_batch_size: 400 # evaluation batch size
5+
num_workers: 10 # number of subprocesses to use for data loading
6+
max_epochs: 24 # number of maximum epochs
7+
use_amp: false # use AMP (Automatic Mixed Precision)
8+
weight_decay: 1e-4
9+
debug: false
10+
train_epoch_length: null
11+
eval_epoch_length: null
12+
filename_prefix: training
13+
n_saved: 2
14+
save_every_iters: 1000
15+
patience: 10
16+
output_dir: ./logs
17+
log_every_iters: 10
18+
lr: 1e-2 # learning rate
19+
model: resnet18 # model name

0 commit comments

Comments
 (0)