Table 2 Hyperparameters tuned for each base TL model.

From: A convolutional recurrent neural network with attention for response prediction to repetitive transcranial magnetic stimulation in major depressive disorder

Model name

Batch size

Learning rate

VGG16

20

1e−5

EfficientNetB0

16

2e−5

Back to article page