参数

取值

输入图像批处理大小

2

初始学习率

权重衰减

训练轮次

104

104

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