neodroidvision.classification.procedures.classification_procedures.pred_target_train_model

neodroidvision.classification.procedures.classification_procedures.pred_target_train_model(model, train_iterator, criterion, optimiser, scheduler, writer, interrupted_path, test_data_iterator=None, num_updates: int = 250000, early_stop=None) → Module

Parameters

• model –

• train_iterator –

• criterion –

• optimiser –

• scheduler –

• writer –

• interrupted_path –

• test_data_iterator –

• num_updates –

• early_stop –

Returns: