| 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 |