Still not working. I dont have much exp with pytorch as m new to it and now m stuck with this. I just wanted to use this model on Android app but now it seems i have to drop this. I followed what they told on pytorch website, load the trained model, trace it to make it executable for android and run it.

import detecto.core
import torch
model = detecto.core.Model.load(model_path, classlist)
example = torch.rand(3, 224, 224)
model_trace = torch.jit.trace(model, example)
model_trace.save(save_path + ‘traced_model’)

I just dont want to retrain thr model, it takes a lot of time but i will have to.