Bhupender Kaushal

Results 4 comments of Bhupender Kaushal

did you resolve it ? Kindly share your solution if possible........

hey @mtailanian were you able to setup the project and train it, if so please do share your frozen pip/ env specs.

Hey, but how did you install cuda toolkit 12.1 with torch 1.13 is that supported ?

import timm import torch # Download and load the pretrained model model = timm.create_model('maxvit_rmlp_small_rw_224.sw_in1k', pretrained=True) model.eval() # Save the model weights as a .pth file torch.save(model.state_dict(), 'maxvit_rmlp_small_rw_224.sw_in1k.pth') print("Model weights saved...