mmpose冻结参数训练,如何添加find_unused_parameters参数

mmpose冻结参数训练,如何添加find_unused_parameters参数


在backbone下方添加 frozen_stages=7即可冻结前7层的参数。要注意对于多卡训练来说还需要添加 find_unused_parameters = True。看图片中代码的位置。
mmpose冻结参数训练,如何添加find_unused_parameters参数_第1张图片

你可能感兴趣的:(python)