When I enabled the rotation, program: "RuntimeError: CUDA error: device-side assert triggered about online-3d-bpp-drl HOT 2 CLOSED

I have the same problem, how did you solve it?

from online-3d-bpp-drl.

Hello. I have a similar problem. Please advise whether I should start or not a new post or not. When I enable rotations the program starts well but then I get the following:
...
enable_rotation: True
use cuda: True
item set: [(2, 2, 2), (2, 2, 3), (2, 2, 4), (2, 2, 5), (2, 3, 2), (2, 3, 3), (2, 3, 4), (2, 3, 5), (2, 4, 2), (2, 4, 3), (2, 4, 4), (2, 4, 5), (2, 5, 2), (2, 5, 3), (2, 5, 4), (2, 5, 5), (3, 2, 2), (3, 2, 3), (3, 2, 4), (3, 2, 5), (3, 3, 2), (3, 3, 3), (3, 3, 4), (3, 3, 5), (3, 4, 2), (3, 4, 3), (3, 4, 4), (3, 4, 5), (3, 5, 2), (3, 5, 3), (3, 5, 4), (3, 5, 5), (4, 2, 2), (4, 2, 3), (4, 2, 4), (4, 2, 5), (4, 3, 2), (4, 3, 3), (4, 3, 4), (4, 3, 5), (4, 4, 2), (4, 4, 3), (4, 4, 4), (4, 4, 5), (4, 5, 2), (4, 5, 3), (4, 5, 4), (4, 5, 5), (5, 2, 2), (5, 2, 3), (5, 2, 4), (5, 2, 5), (5, 3, 2), (5, 3, 3), (5, 3, 4), (5, 3, 5), (5, 4, 2), (5, 4, 3), (5, 4, 4), (5, 4, 5), (5, 5, 2), (5, 5, 3), (5, 5, 4), (5, 5, 5)]
Traceback (most recent call last):
File "main.py", line 258, in
main(args)
File "main.py", line 40, in main
test_model()
File "main.py", line 47, in test_model
unified_test(model_url, config)
File "unified_test.py", line 40, in unified_test
nmodel = nnModel(url, config)
File "model_loader.py", line 16, in init
self._model = self._load_model(url)
File "model_loader.py", line 33, in _load_model
actor_critic.load_state_dict(load_dict)
File "/home/vlad/anaconda3/envs/3D-BPP-DRL/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 Policy:
size mismatch for base.mask.5.weight: copying a param with shape torch.Size([100, 256]) from checkpoint, the shape in current model is torch.Size([200, 256]).
size mismatch for base.mask.5.bias: copying a param with shape torch.Size([100]) from checkpoint, the shape in current model is torch.Size([200]).
size mismatch for dist.linear.weight: copying a param with shape torch.Size([100, 256]) from checkpoint, the shape in current model is torch.Size([200, 256]).
size mismatch for dist.linear.bias: copying a param with shape torch.Size([100]) from checkpoint, the shape in current model is torch.Size([200]).

If I disable rotations from terminal then everything works well. Please advise related to the subject.
Thank you

from online-3d-bpp-drl.