Формы MAT1 и MAT2 не могут быть умножены (64x121 и 7744x64) ⇐ Python

[Anonymous]

Недавно я пытался заставить агента понга с использованием обучения подкреплению, но я продолжаю получать ошибки, независимо от того, что я делаю. Реализация PPO, которую я использую, была первоначально разработана для Cartpole, и я изменил его для работы с Pong.

Код: Выделить всё

with torch.no_grad():
dummy = torch.zeros(1, 1, 96, 96)  # batch=1, grayscale, 96x96
conv_out = self.shared_layers(dummy)
flattened_size = conv_out.view(1, -1).shape[1]

# Inside ActorCriticNetwork.__init__()
with torch.no_grad():
dummy = torch.zeros(1, 1, 96, 96)  # simulate one preprocessed image
conv_out = self.shared_layers(dummy)
flattened_size = conv_out.view(1, -1).shape[1]
< /code>
и как он подключается к линейному слою: < /p>
self.policy_layers = nn.Sequential(
nn.Linear(flattened_size, 64),  # ← uses flattened_size
nn.ReLU(),
nn.Linear(64, action_space_size)
)

Полный код: https://github.com/twhighschooler/pong


Подробнее здесь: https://stackoverflow.com/questions/797 ... nd-7744x64