I would greatly appreaciate it if you can help me to solve the error below
When training  in google colab I get the following:
untimeError                              Traceback (most recent call last)
 in ()
3 for epoch in range(num_epochs):
4     # train for one epoch, printing every 10 iterations
----> 5     train_one_epoch(mask_model, optimizer, data_loader, device, epoch, print_freq=10)
6     # update the learning rate
7     lr_scheduler.step()
9 frames
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py in conv2d_forward(self, input, weight)
340                             _pair(0), self.dilation, self.groups)
341         return F.conv2d(input, weight, self.bias, self.stride,
–> 342                         self.padding, self.dilation, self.groups)
343
344     def forward(self, input):
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same.
I