Hi, I was running the demo code of seqnet:
python demo.py --cfg ~/checkpoints/seqnet/config.yaml --ckpt ~/checkpoints/seqnet/epoch_17.pth

but I got the following error:

Traceback (most recent call last):
File "demo.py", line 89, in
main(args)
File "demo.py", line 63, in main
query_feat = model(query_img, query_target)[0]
File "/home/mmmm/miniconda3/envs/seqnet/lib/python3.7/site-packages/torch/nn/modules/module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "/home/mmmm/repo/SeqNet/models/seqnet.py", line 134, in forward
return self.inference(images, targets, query_img_as_gallery)
File "/home/mmmm/repo/SeqNet/models/seqnet.py", line 117, in inference
box_features = self.roi_heads.box_roi_pool(features, boxes, images.image_sizes)
File "/home/mmmm/miniconda3/envs/seqnet/lib/python3.7/site-packages/torch/nn/modules/module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "/home/mmmm/miniconda3/envs/seqnet/lib/python3.7/site-packages/torchvision/ops/poolers.py", line 141, in forward
sampling_ratio=self.sampling_ratio
File "/home/mmmm/miniconda3/envs/seqnet/lib/python3.7/site-packages/torchvision/ops/roi_align.py", line 70, in roi_align
return _RoIAlignFunction.apply(input, rois, output_size, spatial_scale, sampling_ratio)
File "/home/mmmm/miniconda3/envs/seqnet/lib/python3.7/site-packages/torchvision/ops/roi_align.py", line 25, in forward
output_size[0], output_size[1], sampling_ratio)
RuntimeError: Expected tensor for argument #1 'input' to have the same type as tensor for argument #2 'rois'; but type Variable[CUDAFloatType] does not equal Variable[CUDALongType] (while checking arguments for ROIAlign_forward_cuda) (checkSameType at /pytorch/aten/src/ATen/TensorUtils.cpp:140)

So far I had no idea what caused this error, could you help me with this problem? Thx a lot.