Table 1 Model training environment configuration.
From: Research on dimension measurement algorithm for parcel boxes in high-speed sorting system
Parameter | Configuration |
|---|---|
Programming language | Python |
Deep learning framework | Pytorch1.8.1 |
CPU | Inter(R) Xeon Gold 6248R |
GPU | NVDIA RTX A6000 |
RAM | DDR4 2933 MHz 128 GB |
CUDA | 11.8 |
Image size | \(640 \times 640\) |
Batch size | 16 |
Initial learning rate | 0.01 |
Epoch | 200 |
SGD momentum | 0.937 |
Weight decay | 0.0005 |
