Table 2 Training hyperparameter settings.

From: TCBGY net for enhanced wear particle detection in ferrography using self attention and multi scale fusion

Configuration parameters

Batch size

Epoch

Initial learning rate

Final learning rate

Momentum

Weight decay

Values

32

200

0.01

0.01

0.937

0.0005

Back to article page