Tag: GPYTORCH
Intenté encontrar el código fuente de la atención multihead pero no pude encontrar ningún detalle de implementación. Me pregunto si este módulo solo contiene la parte de atención en lugar de todo el bloque transformador (es decir, no contiene la capa de normalización, la conexión residual y una red neuronal . . . Read more
Estoy leyendo vídeos y los divido en lotes. No puedo ajustar un vídeo completo en mi lote, por lo que los divido usando la función de agrupación. La forma del vídeo es (128,3,224,224), donde 128 son los fotogramas y 3 son los canales (RGB). La forma de salida del cargador . . . Read more
Estoy teniendo problemas para cargar una red neuronal que ha sido entrenada en un nuevo archivo. He guardado la red neuronal en Model.py con torch.save(model.state_dict(), “model.pth”) Ahora, en un nuevo archivo (Test.py) tengo el siguiente código: from Model import NeuralNetwork import torch model = NeuralNetwork() model.load_state_dict(torch.load(model.pth)) model.eval() Sin embargo, cuando . . . Read more
Me gustaría poner algunos tensores en una lista, y sé que si quiero poner la clase nn.Module en una lista, debo usar ModuleList para envolver esa lista. Entonces, ¿existe algo como ‘TensorList’ en pytorch, que debo usar para envolver la lista que contiene tensores?
He encontrado un código que convierte directamente el modelo de PyTorch a ONNX: import torch.onnx torch.onnx.export( model, input, “model.onnx”, export_params=True, opset_version=10 ) Pero arroja una advertencia (“UserWarning”) la mayoría de las veces: /usr/local/lib/python3.7/dist-packages/torch/nn/functional.py:2359: UserWarning: ‘floordiv’ está descontinuado y su comportamiento cambiará en una versión futura de PyTorch. Actualmente redondea hacia . . . Read more