Open zhuyusenCV opened 1 year ago
When I'm done training on my own dataset,there was a problem while testing。Can you tell me why?
net = GateNet().cuda() TypeError: init() missing 2 required positional arguments: 'block1' and 'layers'
Replace the code using:
net = GateNet(Bottleneck, [3,4,6,3]).cuda()
Just like the train.py
train.py
When I'm done training on my own dataset,there was a problem while testing。Can you tell me why?
net = GateNet().cuda() TypeError: init() missing 2 required positional arguments: 'block1' and 'layers'