can't transfer the info dictionary {} generated by env.step() to inference queue about torchbeast HOT 2 CLOSED

Hey mschen97,

Thanks for your interest in TorchBeast, as well as for your interest in SC2 with which I also have some history.

Our queues can only handle tensor data. That means the arguments are either tensors, or lists or dicts with string keys. If they are lists or dicts, the contents (values) need adhere to the same constraint. Nested lists/dicts are possible, but in the end all the "leaf" entries need to be tensors/np.arrays. The number of objects doesn't matter though. However, care needs to be taken that each environment step returns the same number of tensors and that these tensors are compatible (batchable).

There's no tensor type restriction as far as I remember.

Perhaps if you could share some short example code I could help you further?

from torchbeast.

Thanks for your replay, I have solved this problem by changing my fifth output to tensor and then form a tuple with the first output to output at the first position.Thanks!

from torchbeast.