tensorflow / tensorflow/models
Transfer learning code not working when loading dataset
Open
@rachellj218 is already working on this.
Since Sep 14, 2021.
models:research:odapi
type:bug
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
Prerequisites
- I am using the latest TensorFlow Model Garden release and TensorFlow 2.
- I am reporting the issue to the correct repository
- I checked to make sure that this issue has not been filed already.
The entire URL of the file I am using
https://github.com/tensorflow/models
Description of the bug
From the file model_lib_v2.py the transfer learning process seems to enter in an infinite loop in
features, labels = iter(input_dataset).next()
at the ligne 159
System information
- OS Platform and Distribution (e.g., Window 10):
- Mobile device name if the issue happens on a mobile device:
- TensorFlow installed from (source or binary):
- TensorFlow version : 2
- Python version: 3.7
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.