poets-ai / elegy

A High Level API for Deep Learning in JAX
https://poets-ai.github.io/elegy/
MIT License
469 stars 32 forks source link

ValueError: The `target` has to be an integer tensor. #241

Open mwitiderrick opened 2 years ago

mwitiderrick commented 2 years ago

What is the right way to use dataloaders from PyTorch and generators from TensorFlow?

This gave the above error

history = model.fit(
    inputs=train_loader,
    epochs=100,
    steps_per_epoch=200,
    validation_data=validation_loader,
    shuffle=True,
    callbacks=[eg.callbacks.ModelCheckpoint("models/high-level", save_best_only=True)],
)