Problem with torch.FloatTensor and torch.cuda.FloatTensor

[duc-anh-2002]

Thank you for your repository, at this time, I have the problem as follows:
" File "main_debug.py", line 495, in main
train_stats = train_one_epoch(
File "/home/ubuntu/workspace/mamba/simba/classification/engine.py", line 48, in train_one_epoch
outputs = model(samples)
File "/home/ubuntu/workspace/envs/simba/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "/home/ubuntu/workspace/mamba/simba/classification/simba_debug.py", line 603, in forward
x, H, W = self.forward_embeddings(x)
File "/home/ubuntu/workspace/mamba/simba/classification/simba_debug.py", line 659, in forward_embeddings
x, H, W = patch_embed(x)
File "/home/ubuntu/workspace/envs/simba/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "/home/ubuntu/workspace/mamba/simba/classification/simba_debug.py", line 477, in forward
x = self.conv(torch.FloatTensor(x))
File "/home/ubuntu/workspace/envs/simba/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "/home/ubuntu/workspace/envs/simba/lib/python3.8/site-packages/torch/nn/modules/container.py", line 139, in forward
input = module(input)
File "/home/ubuntu/workspace/envs/simba/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "/home/ubuntu/workspace/envs/simba/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 457, in forward
return self._conv_forward(input, self.weight, self.bias)
File "/home/ubuntu/workspace/envs/simba/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 453, in _conv_forward
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor"
Could you help me to fix this bug, thank you