Skip to content

Commit

Permalink
a little change
Browse files Browse the repository at this point in the history
  • Loading branch information
Felixvillas committed Sep 25, 2022
1 parent c3c847b commit 9677b43
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion model.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ def get_batch_value(self, obs):

q_value = []
rnn_value = torch.zeros((batch_size*num_agents, self.net_embed_dim), dtype=torch.float32, device=device)
obs = obs.clone().transpose(1, 2).reshape(batch_size*num_agents, timesteps, obs_dim)
obs = obs.transpose(1, 2).reshape(batch_size*num_agents, timesteps, obs_dim)
for t in range(timesteps):
'''
note: (batch_size, num_agents, dim) --> (batch_size*num_agents, dim) [###By tensor.reshape]
Expand Down

0 comments on commit 9677b43

Please sign in to comment.