I’ experimenting with converting Pytorch models to ONNX using this repository(repo) - It is basically implementation of SSD.
Tracer argues at this script that computes coordinates for prior boxes.(script)
Specifically this code - where tensor is converted into python range:
for i, j in product(range(f), repeat=2): f_k = self.image_size / self.steps[k] # unit center x,y cx = (j + 0.5) / f_k cy = (i + 0.5) / f_k
After that the whole process hangs. Any idea what might cause this or suggestions how to rewrite this part?
Thanks a lot for your help!