es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Tag: GPYTORCH

Utilización de la CPU al utilizar ray y torch.

Utilizo ray y torch en mi código y configuro un núcleo de CPU para cada actor remoto de ray para computar el gradiente (usando el paquete torch). Pero encuentro que la utilización de CPU del actor puede llegar a ser de hasta 300% en algunas ocasiones. Esto parece ser imposible . . . Read more

Forma eficiente de enviar un tensor de datos a través de una función sin un bucle.

Tengo una red neuronal (mi función): import torch import torch.nn as nn import torch.optim as optim softplus = torch.nn.Softplus() class Model(nn.Module): def __init__(self): super(Model, self).__init__() self.fc1 = nn.Linear(4, 10) self.fc2 = nn.Linear(10, 10) self.fc3 = nn.Linear(10, 1) def forward(self, x): x = torch.relu(self.fc1(x)) x = torch.relu(self.fc2(x)) x = softplus(self.fc3(x)) return . . . Read more