CameraRadarFusionNet
CameraRadarFusionNet copied to clipboard
multi_gpu_model AttributeError: 'NoneType' object has no attribute 'fusion_blocks'
When multi_gpu=0, the network can be trained smoothly,
but when I change multi_gpu=8, I get the error bellow:
It seems likely the cfg cannot be passed to backbone_retinanet.
it seems like in train_crfnet.py line 131,
backbone_retinanet(num_classes, num_anchors=num_anchors, modifier=modifier, inputs=inputs, distance=distance)
cfg=cfg is missed
File "train_crfnet.py", line 424, in
I solved it. I think authors work lack a var of "cfg" in train_crfnet in line 134:
the previous line:
backbone_retinanet(num_classes, num_anchors=num_anchors, modifier=modifier, inputs=inputs, distance=distance),weights=weights, skip_mismatch=True, config=copy.deepcopy(cfg),num_classes=num_classes)
after modifying(add "cfg = cfg")
backbone_retinanet(num_classes, num_anchors=num_anchors, modifier=modifier, inputs=inputs, distance=distance, **cfg = cfg**), weights=weights, skip_mismatch=True, config=copy.deepcopy(cfg), num_classes=num_classes)
@guiyuliu