여러 Area를 동시에 학습

안녕하세요,

Drone 강의 부분에서, Area를 GridWorld에서 했던 것처럼 여러 Area를 복사하여 한 씬에 넣고 동시에 학습시키면 step이 빨리 오르는 효과를 볼 수 있을 것 같다고 생각했습니다.

한 area를 여러 개 복사한 뒤에 학습을 진행하려 했는데, 다음과 같은 오류로 잘 되지 않았습니다.

어떻게 해결해 볼 수 있을까요?

critic_loss = F.mse_loss(target_q, q)

Traceback (most recent call last):

File "DDPG.py", line 240, in <module>

actor_loss, critic_loss = agent.train_model()

File "DDPG.py", line 144, in train_model

reward = np.stack([b[2] for b in batch], axis=0)

File "<__array_function__ internals>", line 6, in stack

File "C:\Users\anaconda3\envs\mlagent2\lib\site-packages\numpy\core\shape_base.py", line 426, in stack

raise ValueError('all input arrays must have the same shape')

ValueError: all input arrays must have the same shape