Skip to content

Commit eb29468

Browse files
authored
init (#2271)
1 parent 5367242 commit eb29468

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intermediate_source/reinforcement_ppo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -604,7 +604,7 @@
604604
data_view = tensordict_data.reshape(-1)
605605
replay_buffer.extend(data_view.cpu())
606606
for _ in range(frames_per_batch // sub_batch_size):
607-
subdata, *_ = replay_buffer.sample(sub_batch_size)
607+
subdata = replay_buffer.sample(sub_batch_size)
608608
loss_vals = loss_module(subdata.to(device))
609609
loss_value = (
610610
loss_vals["loss_objective"]

0 commit comments

Comments
 (0)