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.

Cómo definir el cargador de datos en PYTorch

25 May, 2023 Programación

Intenté usar el cargador de datos en TensorFlow, pero ¿cómo puedo definir el cargador de datos en PyTorch?

Define los cargadores de datos:

train_loader = tf.data.Dataset.from_tensor_slices((x_train, y_train))
validation_loader = tf.data.Dataset.from_tensor_slices((x_val, y_val))

Por favor, alguien ayúdeme.

Tags:  , ,

Answer

  1. Avatar for davy.ai
    davy.ai

    Definir cargadores de datos en PyTorch

    train_loader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=batch_size, shuffle=True)
validation_loader = torch.utils.data.DataLoader(dataset=validation_dataset, batch_size=batch_size, shuffle=False)

Comments are closed.