Tengo una instancia de notebook ml.t3.2xlarge en AWS Sagemaker y quiero entrenar una red neuronal utilizando pytorch. El siguiente comando retorna “False”: torch.cuda.is_available() ¿Hay algo que necesite hacer para activar la GPU? Cuando ejecuto el siguiente comando: ! nvidia-smi Luego dice “NVIDIA-SMI ha fallado porque no pudo comunicarse con el . . . Read more
¿Cambia el número de parámetros y las operaciones en coma flotante por segundo (FLOPS) al convertir un modelo de PyTorch a formato ONNX o TensorRT?
Estoy intentando convertir el código del modelo de CNN de Keras a Pytorch. Aquí está la capa secuencial de Keras: model = Sequential() model.add(Conv2D(filters=64, kernel_size=(3,3), activation=”relu”, input_shape=(28,28,1))) model.add(Conv2D(filters=64, kernel_size=(3,3), activation=”relu”)) model.add(MaxPooling2D(pool_size=(2,2))) model.add(BatchNormalization()) model.add(Conv2D(filters=128, kernel_size=(3,3), activation=”relu”)) model.add(Conv2D(filters=128, kernel_size=(3,3), activation=”relu”)) model.add(MaxPooling2D(pool_size=(2,2))) model.add(BatchNormalization()) model.add(Conv2D(filters=256, kernel_size=(3,3), activation=”relu”)) model.add(MaxPooling2D(pool_size=(2,2))) model.add(Flatten()) model.add(BatchNormalization()) model.add(Dense(512, activation=”relu”)) model.add(Dense(10, activation=”softmax”)) . . . Read more
Cuando paso una imagen y una máscara a albumentations.Normalize(mean, std). ¿Cómo puedo incorporar esto? ¿Debo agregarlo manualmente en el conjunto de datos? ¡Agradecido por cualquier consejo que tenga!
He entrenado un modelo de clasificación DistilBERT utilizando Huggingface y el modelo parece funcionar bien, con una pérdida de alrededor de 0.3 después de probar el mejor modelo después del entrenamiento con el siguiente código: trainer.evaluate() Sin embargo, al intentar cargar el modelo y evaluar la pérdida nuevamente en el . . . Read more