参数 | 取值 |
输入图像批处理大小 | 2 |
初始学习率 权重衰减 训练轮次 | 10−4 10−4 200 |
参数 | 实验配置 |
系统 | ubuntu18.04 |
CPU | Intel(R) Xeon(R) Platinum 8358P CPU @ 2.60GHz |
GPU | NVIDIA GeForce RTX 3090 24G |
深度学习框架 开发环境 | Pytorch torch1.9.0,cuda11.1,Python3.8 |