配置项 | 配置参数 |
内存 | 16GB |
显卡(显存) | NVIDIA RTX3060(8GB) |
处理器(CPU) | Intel i5-12490F |
深度学习框架 | Pytorch |
CUDA版本 | 11.7 |
输入图片尺寸 | 300 × 300 |
权重衰减参数 | 0.1 |
batchsize | 16 |
初始学习率 | 0.001 |
MAX_ITER | 120000 |