This repository has been archived by the owner on Oct 9, 2023. It is now read-only.
Add support for passing the output to use to the predict
method
#920
Labels
Milestone
This should be an
output
argument that can take either a string (the name of the output) or anOutput
object to use (e.g. a custom one).This should also be added to
flash.Trainer.predict
The text was updated successfully, but these errors were encountered: