nn.Parameter факела () в Керасе - PullRequest
Новая
       71

nn.Parameter факела () в Керасе

0 голосов
/ 16 марта 2020

Я хочу реализовать nn.Parameter() в Керасе, на самом деле эта функция возвращает выходные данные, которые отличаются при каждом прямом проходе, и я до сих пор не знаю, откуда берутся эти значения

, и я хочу реализовать этот код факела в Код Keras, но я решаю эту проблему

вот полный код: 

def __init__(self, input_shape, n_actions):
        super(A2C_policy, self).__init__()

        self.lp = nn.Sequential(
            nn.Linear(input_shape[0], 32),
            nn.ReLU(),
            nn.Linear(32, 32),
            nn.ReLU())

        self.mean_l = nn.Linear(32, n_actions[0])
        self.mean_l.weight.data.mul_(0.1)

        self.var_l = nn.Linear(32, n_actions[0])
        self.var_l.weight.data.mul_(0.1)

        self.logstd = nn.Parameter(torch.zeros(n_actions[0]))
...