deepextractor.training.train_fn¶

Module Contents¶

deepextractor.training.train_fn.train_fn(loader, model, model_name, optimizer, loss_fn, scaler, device)[source]¶: Train the model for one epoch and return average losses.