pytorch中优化器optimizer.param_groups[0]是什么意思

2022-09-02 21:01:23 浏览数 (1)

optimizer.param_groups:是长度为2的list,其中的元素是2个字典; optimizer.param_groups[0]:长度为6的字典,包括[‘amsgrad’, ‘params’, ‘lr’, ‘betas’, ‘weight_decay’, ‘eps’]这6个参数 optimizer.param_groups[1]:好像是表示优化器的状态的一个字典

0 人点赞