Table 1 ResNet18 DL model Hyper-parameters.

From: Enhanced classification prostate cancer based on generative adversarial networks and integrated deep learning with vision transformer models

Model

Learning Rate

Epoch

Optimizer

Batch Size

Weight Decay

DensNet121

0.001

200

SGD

32

\(\:{10}^{-5}\)

EfficientNet-B4

0.003

150

SGD

32

\(\:{10}^{-3}\)

ResNet18

0.002

400

SGD

32

\(\:{10}^{-4}\)

Back to article page