File size: 266 Bytes
0b11a42 |
1 2 3 4 5 6 7 8 9 |
skorch_model:
_target_: transforna.Net
module: transforna.GeneEmbeddModel
criterion: transforna.LossFunction
max_epochs: 0 #infered from task specific train config
optimizer: torch.optim.AdamW
device: cuda
batch_size: 64
iterator_train__shuffle: True |