Table 1 Hyperparameters used for the segmentation models.
From: A high-precision segmentation method based on UNet for disc cutter holder of shield machine
Learning rate | 0.01 |
|---|---|
Optimizer | SGD |
Momentum | 0.9 |
Loss function | Cross-entropy |
Batch size | 16 |
Number of epochs | 100 |
