MaxViT

MaxViT copied to clipboard

Thanks to the authors for sharing Maxvit open source, I really enjoyed this project and studied it for a few days. However, I didn't understand this part of the work...

DDxk369

error: RuntimeError: Tensor for argument #2 'weight' is on CPU, but expected it to be on GPU (while checking arguments for cudnn_batch_norm)

6

Thank you for your work. I import the model, and push it to gpu, but the error happen: RuntimeError: Tensor for argument #2 'weight' is on CPU, but expected it...

Geek-lixiang

Hi Chrstoph, thanks for code skeleton for MaxViT paper. I checked the number of parameters of your code and paper, and both seems to be difference. MaxViT tiny give 24M...

saarthak-kapse

In this line of code output = self.main_path(input)：line 69, in forward Error RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and...

LvGuangzu

model = max_vit_base_224(num_classes=10) print(model) ----------------------------------------------------------------------------------------------------------- terminal output: (stem): Sequential( (0): Conv2d(3, 64, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1)) (1): GELU() (2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (3):...

jasnei

changed module names in the timm library. Replaced `efficentnet_blocks` with `_efficientnet_blocks`

Mr-TalhaIlyas