Table 2 Training parameter settings.

From: Research on steel surface defect classification method based on deep learning

Training parameters

Value

Learn rate

0.01

Momentum

0.937

weight_decay

0.0005

image_size

224 × 224

Epochs

100

Back to article page