HigherHRNet代码复现问题集(assert isinstance(orig, torch.nn.Module))

1. 报错：assert isinstance(orig, torch.nn.Module)

出现在tools/dist_train.py文件内。
根据github说明，输入：

python tools/dist_train.py \
    --cfg experiments/coco/higher_hrnet/w32_512_adam_lr1e-3.yaml 

可能会出现

原因：tensorboardX库出了问题。
解决方式1：
注释代码：

#writer_dict['writer'].add_graph(model, (dump_input, ))

但这样，无法保存tensorboardX日志

解决方案2：
重新下载特定版本的torch, torchvision,tensorboardX,可解决这个问题。
论文作者的版本肯定可以避免这些错误。但具体版本不知道。

我个人试出来的版本：

torch==1.8
torchvision == 0.9.0
tensorboardX ==2.2