You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Probably we can make loss_creator optional. If so, they inputs and targets would all be fed into the model, and user should define a function to get the loss from the model forward result (and then to combine with loss.backward)?
When train yolox on orca, there is no need to pass the 'loss creator' as parameter to initialize Estimator and calculate the loss after forward(). Because yolox can directly get the loss after forward().
https://github.com/Megvii-BaseDetection/YOLOX/blob/main/yolox/core/trainer.py#L106
The same is true when evaluate yolox on orca, there is no need to calculate the loss after forward()
The text was updated successfully, but these errors were encountered: