Вопросы с тегом «torch»

Я запутался в методе view()в следующем фрагменте кода. 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))) x = …

208 python  memory  pytorch  torch  tensor 

Я читал документацию PyTorch и нашел пример, в котором они пишут gradients = torch.FloatTensor([0.1, 1.0, 0.0001]) y.backward(gradients) print(x.grad) где x была начальной переменной, из которой был построен y (3-вектор). Вопрос в том, каковы аргументы 0,1, 1,0 и 0,0001 тензора градиентов? Документация по этому поводу не очень ясна.

112 neural-network  gradient  pytorch  torch  gradient-descent 

Я пытаюсь пройти процесс установки Github на MacOs Catalina Первый шаг - выполнить в Терминале: cd ~/ curl -s https://raw.githubusercontent.com/torch/ezinstall/master/install-deps | bash Но я получаю ошибку: Error: caskroom/cask was moved. Tap homebrew/cask-cask instead. Тогда я попробовал: brew tap caskroom/cask Но я получил ту же ошибку. Есть ли у вас какие-либо …

11 macos  homebrew  torch  macos-catalina  homebrew-cask