Ayoosh Kathuria

Umm, That's an error on my part. I've fixed it in the tutorial. If you see the definition of `forward`, ``` def forward(self, x, CUDA): modules = self.blocks[1:] outputs =...

At what line are you getting this error? This is error is prolly being generated by trying to add two tensors one of which is a cuda tensors and other...

Hey? Did you solve your issue?

@RyanCV @noirblack Instead of modifying the code inside the `create_module` function simply do, ``` CUDA = torch.cuda.is_avaliable() model = Darknet("cfg/yolov3.cfg") if CUDA: model = model.cuda() inp = get_test_input() pred =...