Perguntas com a marcação «torch»

Estou confuso sobre o método view()no seguinte trecho de código. class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5) self.pool = nn.MaxPool2d(2,2) self.conv2 = nn.Conv2d(6, 16, 5) self.fc1 = nn.Linear(16*5*5, 120) self.fc2 = nn.Linear(120, 84) self.fc3 = nn.Linear(84, 10) def forward(self, x): x = self.pool(F.relu(self.conv1(x))) x = self.pool(F.relu(self.conv2(x))) …

205 python  memory  pytorch  torch  tensor 

Estou lendo a documentação do PyTorch e encontrei um exemplo onde eles escrevem gradients = torch.FloatTensor([0.1, 1.0, 0.0001]) y.backward(gradients) print(x.grad) onde x era uma variável inicial, a partir da qual y foi construído (um 3-vetor). A questão é: quais são os argumentos de 0,1, 1,0 e 0,0001 do tensor de …

112 neural-network  gradient  pytorch  torch  gradient-descent 

Eu tento passar pelo processo de instalação do Github no MacOs Catalina O primeiro passo é executar no Terminal: cd ~/ curl -s https://raw.githubusercontent.com/torch/ezinstall/master/install-deps | bash Mas eu recebo o erro: Error: caskroom/cask was moved. Tap homebrew/cask-cask instead. Então eu tentei: brew tap caskroom/cask Mas eu recebi o mesmo erro. …

11 macos  homebrew  torch  macos-catalina  homebrew-cask