Research Article

Transfer Learning with Feature Extraction Modules for Improved Classifier Performance on Medical Image Data

Table 3

Training parameters.

ParametersValue

Momentum0.95
Dropout0.3
Learning rate0.0001
LossBinary cross-entropy, categorical cross-entropy
OptimizerSGD
Epochs30
Batch size16