Граница последовательных параметров ANN Pytorch - PullRequest
Новая
       0

Граница последовательных параметров ANN Pytorch

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

Как найти нижний и верхний предел (границу) значений, взятых параметрами (вес и смещение) в нейронной сети (последовательной) в pytorch? например, как [-1 1] или [-0,5 0,5]

Правильно ли я понял, что пределы значений параметров изменяются в пиктограмме в зависимости от количества входов и выходов? https://pytorch.org/docs/stable/nn.init.html

    class CartPoleAI(nn.Module):
        def __init__(self,inputsize,hiddensize,outputsize):
            super().__init__()
            self.fc = nn.Sequential(
                        nn.Linear(inputsize,hiddensize, bias=True),
                        nn.ReLU(),
                        nn.Linear(hiddensize,outputsize, bias=True),
                        nn.Softmax(dim=1)
                        )


        def forward(self, inputs):
            x = self.fc(inputs)
            return x


     agent = CartPoleAI(inputsize,hiddensize,outputsize)
     for param in agent.parameters():
        param.requires_grad = False
        print(param)
...