model: The original PyTorch model. input_keys: List of input names as keys for the input dictionary.
(self, actor)
| 100 | |
| 101 | class PPOEncoderWrapper(nn.Module): |
| 102 | def __init__(self, actor): |
| 103 | """ |
| 104 | model: The original PyTorch model. |
| 105 | input_keys: List of input names as keys for the input dictionary. |
| 106 | """ |
| 107 | super(PPOEncoderWrapper, self).__init__() |
| 108 | self.actor = actor |
| 109 | |
| 110 | def forward(self, inputs): |
| 111 | """ |