start_train error #54
Description
help!help! /aten/src/ATen/native/cuda/ScatterGatherKernel.cu:365: operator(): block: [36,0,0], thread: [104,0,0] Assertion idx_dim >= 0 && idx_dim < index_size && "index out of bounds" failed.
{{ message }}
help!help! /aten/src/ATen/native/cuda/ScatterGatherKernel.cu:365: operator(): block: [36,0,0], thread: [104,0,0] Assertion idx_dim >= 0 && idx_dim < index_size && "index out of bounds" failed.