You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
20 lines
295 B
YAML
20 lines
295 B
YAML
# Training params
|
|
epoch : 10
|
|
batch_size : 4
|
|
num_workers: 8
|
|
limit : 5
|
|
|
|
# Optimizer
|
|
optimizer :
|
|
name : torch.optim.Adam
|
|
lr : 0.0001
|
|
|
|
# Criterion
|
|
loss_function :
|
|
name : torch.nn.CrossEntropyLoss
|
|
|
|
# Scheduler
|
|
scheduler :
|
|
name : torch.optim.lr_scheduler.StepLR
|
|
gamma : 0.01
|
|
step_size : 10 |