RuntimeError: expected backend CPU and dtype Float but got backend CPU and dtype Long

[zihaozhang9]

A ERROR in encoder.py#L59
error log:
RuntimeError: expected backend CPU and dtype Float but got backend CPU and dtype Long

[zihaozhang9]

I fixed this error!

xy = (xy*grid_size).view(fm_h,fm_w,1,2).expand(fm_h,fm_w,9,2)
change to :
xy = (xy.float()*grid_size).view(fm_h,fm_w,1,2).expand(fm_h,fm_w,9,2)