DouZero
/
Douzero_Resnet
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修bug

This commit is contained in:
zhiyang7 2021-12-10 16:28:03 +08:00
parent 05aa179ba6
commit 6dcfe074de
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
douzero/dmc/utils.py
View File

@ -53,10 +53,16 @@ def get_batch(b_queues, position, flags, lock):
buffer = [] buffer = []
while len(buffer) < flags.batch_size: while len(buffer) < flags.batch_size:
buffer.append(b_queue.get()) buffer.append(b_queue.get())
if flags.old_model and position != 'bidding':
batch = { batch = {
key: torch.stack([m[key] for m in buffer], dim=1) key: torch.stack([m[key] for m in buffer], dim=1)
for key in ["done", "episode_return", "target", "obs_z", "obs_x_batch", "obs_x_no_action", "obs_type"] for key in ["done", "episode_return", "target", "obs_z", "obs_x_batch", "obs_x_no_action", "obs_type"]
} }
else:
batch = {
key: torch.stack([m[key] for m in buffer], dim=1)
for key in ["done", "episode_return", "target", "obs_z", "obs_x_batch", "obs_type"]
}
del buffer del buffer
return batch return batch