deepextractor.training.train_fn
===============================

.. py:module:: deepextractor.training.train_fn




Module Contents
---------------

.. py:function:: train_fn(loader, model, model_name, optimizer, loss_fn, scaler, device)

   Train the model for one epoch and return average losses.