Open SpiritBear000 opened 5 years ago
when i run train_baseline.py
########## error log ############
Traceback (most recent call last): File "/Person-reid-GAN-pytorch/train_baseline.py", line 371, in num_epochs=130) File "/Person-reid-GAN-pytorch/train_baseline.py", line 251, in train_model loss = criterion(outputs,labels,flags) File "/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(*input, **kwargs) File "/Person-reid-GAN-pytorch/train_baseline.py", line 178, in forward return loss.mean() RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generated/../THCReduceAll.cuh:339 /pytorch/aten/src/THC/THCTensorScatterGather.cu:97: void THCudaTensor_gatherKernel(TensorInfo<Real, IndexType>, TensorInfo<Real, IndexType>, TensorInfo<long, IndexType>, int, IndexType) [with IndexType = unsigned int, Real = float, Dims = 2]: block: [0,0,0], thread: [10,0,0] Assertion indexValue >= 0 && indexValue < src.sizes[dim] failed. /pytorch/aten/src/THC/THCTensorScatterGather.cu:97: void THCudaTensor_gatherKernel(TensorInfo<Real, IndexType>, TensorInfo<Real, IndexType>, TensorInfo<long, IndexType>, int, IndexType) [with IndexType = unsigned int, Real = float, Dims = 2]: block: [0,0,0], thread: [13,0,0] Assertion indexValue >= 0 && indexValue < src.sizes[dim] failed.
indexValue >= 0 && indexValue < src.sizes[dim]
Hey, did you got the solution I am also facing the same issue while training GAN!
when i run train_baseline.py
########## error log ############
Traceback (most recent call last): File "/Person-reid-GAN-pytorch/train_baseline.py", line 371, in
num_epochs=130)
File "/Person-reid-GAN-pytorch/train_baseline.py", line 251, in train_model
loss = criterion(outputs,labels,flags)
File "/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call
result = self.forward(*input, **kwargs)
File "/Person-reid-GAN-pytorch/train_baseline.py", line 178, in forward
return loss.mean()
RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generated/../THCReduceAll.cuh:339
/pytorch/aten/src/THC/THCTensorScatterGather.cu:97: void THCudaTensor_gatherKernel(TensorInfo<Real, IndexType>, TensorInfo<Real, IndexType>, TensorInfo<long, IndexType>, int, IndexType) [with IndexType = unsigned int, Real = float, Dims = 2]: block: [0,0,0], thread: [10,0,0] Assertion
indexValue >= 0 && indexValue < src.sizes[dim]
failed. /pytorch/aten/src/THC/THCTensorScatterGather.cu:97: void THCudaTensor_gatherKernel(TensorInfo<Real, IndexType>, TensorInfo<Real, IndexType>, TensorInfo<long, IndexType>, int, IndexType) [with IndexType = unsigned int, Real = float, Dims = 2]: block: [0,0,0], thread: [13,0,0] AssertionindexValue >= 0 && indexValue < src.sizes[dim]
failed.