Traceback (most recent call last):


 "generate-embeddings.py", line 34, in <module>

model.load_state_dict(reloaded['model'])


 "/share/pkg.7/pytorch/1.1/install/3.7/lib/python3.7/site-packages/torch/nn/modules/module.py", line 777, in load_state_dict

self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for TransformerModel:

Traceback (most recent call last):


 "/home/ruinianxu/miniconda2/envs/CenterNet/lib/python3.6/threading.py", line 916, in _bootstrap_inner

self.run()


 "/home/ruinianxu/miniconda2/envs/CenterNet/lib/python3.6/threading.py", line 864, in run

self._target(*self._args, **self._kwargs)


 "/home/ruinianxu/IVA_Lab/Project/CenterNet-master/train.py", line 54, in pin_memory

data["ys"] = [y.pin_memory() for y in data["ys"]]


 "/home/ruinianxu/IVA_Lab/Project/CenterNet-master/train.py", line 54, in <listcomp>

data["ys"] = [y.pin_memory() for y in data["ys"]]
RuntimeError: cuda runtime error (59) : device-side assert triggered at /opt/conda/conda-bld/pytorch_1532581333611/work/aten/src/THC/THCCachingHostAllocator.cpp:271
