tensorflow / tensorflow/models
ResNet50V1 classification checkpoint not loaded when training Faster-RCNN from scratch on custom dataset
Open
@pkulzc is already working on this.
Since Aug 31, 2020.
models:research:odapi
type:bug
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
I am trying to use this config file for training on a different dataset.
My tensorflow version is 2.3 (stable).
In order to use the classification checkpoint of ResNet50V1, I did the following:
import tensorflow as tf
model = tf.keras.applications.ResNet50(weights='imagenet')
checkpoint = tf.train.Checkpoint(model=model)
checkpoint.save(<CKPT_PATH>)
Then I provide the path to this checkpoint file in the config file. On running the training using model_main_tf2.py, I receive the same message as has been previously referenced in #8892 .
How can this issue be resolved ? Am I making a mistake here ?
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.