NVIDIA / NVIDIA/DALI

TF Keras Multiple inputs/outputs

Open
#2,700 7 comments 0 reactions 1 assignee View on GitHub

@JanuszL is already working on this.

Since Mar 11, 2022.

enhancement question
Dominant language
C++
Stars
5.8k
Forks
678
Avg merge
3d 1h
Merged PRs (30d)
27

Description

tf.keras allows you to define models with multiple inputs and outputs, but in order to use the standard keras training loops, you need to provide a tf.Dataset which returns either 2 tuples or 2 dictionaries.

I can see how to create a tf.Dataset with DALI that returns more than 2 tensors, but I don't see how to return tuples or dictionaries. Is there a way to do that?

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.