feng3245

> > I have encountered the same problem as you,after changed the third line "grid_size = inp_dim // stride" to "grid_size = prediction.size(2)" in function predict_transform,the problem fixed. > >...

Will resolve conflicts and re-issue pull request

> @feng3245 Just fork this repo. It looks like this feature will never be merged. Someone already created a beta extension out of this check point. Frankly this project is...