Tag: GPYTORCH
Tengo dos matrices A de tamaño (16, 8, 400) y B de tamaño (16, 50, 400); donde 16 es el tamaño del lote. Ahora lo que necesito hacer es lo siguiente: Para cada lote en A, quiero calcular el producto interno de cada fila en un solo lote de A . . . Read more
Generé archivos ckpt con ModelCheckpoint de Pytorch Lightning (save_last=True) en mi cluster que utiliza Linux. En el propio cluster puedo cargarlos sin problemas, pero en mi máquina Windows no puedo y obtengo este error: AttributeError: El módulo ‘torch.distributed.rpc’ no tiene el atributo ‘RRef’ Realmente necesito ayuda, ya que tengo un . . . Read more
Estoy trabajando en tareas de super resolución de imágenes con EDSR como modelo base. Siguiendo a EDSR, no estoy utilizando ninguna capa de batch-norm en mi modelo. De repente, se me ocurrió una pregunta estúpida sobre los tamaños de lote. Actualmente, estoy entrenando mi modelo con un tamaño de lote . . . Read more
Quiero convertir mi modelo de pyTorch (.pth) a ONNX. Este es mi código. import torch import torchvision dummy_input = torch.randn(1,3,224,224,device = “cuda”) model = torch.load(“/home/selfsup/moco/r50_v2/epoch_200.pth”) torch.onnx.export(model,dummy_input,”LENet.onnx”,verbose = True) Sin embargo, obtengo un error que dice lo siguiente. Traceback (most recent call last): File “tengine_convert.py”, line 5, in <module> torch.onnx.export(model,dummy_input,”LENet.onnx”,verbose = . . . Read more
Estoy tratando de verificar el hecho de que un argumento dado sea una pérdida predeterminada de PyTorch (criterio) y optimizar en python para asegurarme de que puedo realizar las siguientes operaciones: loss = criterion(y_pred, y_batch) optimizer.zero_grad() loss.backward() optimizer.step() En última instancia, solo quiero verificar algo como: assert isinstance(criterion, torch.loss) assert . . . Read more