Fix the issue RuntimeError: CUDA error: device-side assert triggered during run this demo py file this PR updating is related to issue: issue42 ...