Efficientnet trained with noisystudent
marcelklehr opened this issue · 4 comments
Is your feature request related to a problem? Please describe.
NoisyStudent is a training method that achieves better results.
Describe the solution you'd like
It would be cool if you could also offer NoisyStudent variants of EfficientNet
thanks, @marcelklehr for open this issue
try follow this repo https://github.com/ntedgi/efficientnet-tensorflowjs-binaries
it contains node-efficientnet tensor flow model there is a simple example on how to transfer tf-model
after you finish transfer you can load it here in the same model
Would you be willing to accept a PR that adds NoisyStudent checkpoints?
Definitely and if you need help I would be happy to help
Before you start modifying the code base try to load your model with this example
‘const path = require("path");
const {
EfficientNetCheckPointFactory,
EfficientNetCheckPoint
} = require("node-efficientnet");
const model = await EfficientNetCheckPointFactory.create(
EfficientNetCheckPoint.B7,
{
localModelRootDirectory: path.join(__dirname, "local_model")
}
);
const path2image = "...";
const topResults = 5;
const result = await model.inference(path2image, {
topK: topResults,
locale: "zh"
});’