Open marcelklehr opened 3 years ago
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" });’
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