amna-ai

I downloaded the whole repo, I got it working but training the model made my laptop stuck. Apparently there's not enough RAM. I was thinking if I could download the...

Did you find any help?

Can you help me the training the model on the RGB? While using the pretrained weights and adding only the top layer for training, I am getting the accuracy of...

Anyone its urgent

I don't know if hardware is causing this result. Did you find any other reason?

I am facing similar issue. I have changed the code to thid snapshot = torch.load(model_weight, map_location=torch.device('cpu')) but still getting the same error you got.