train

train(data: ~local2global_embedding.classfication.ClassificationProblem, model: ~torch.nn.modules.module.Module, num_epochs, batch_size, lr=0.01, batch_logger=<function <lambda>>, epoch_logger=<function <lambda>>, device=None, epsilon=1, alpha=0, beta=0, weight_decay=0.01, decay_lr=False, xi=1e-06, vat_it=1, teacher_alpha=0, beta_1=0.9, beta_2=0.999, adam_epsilon=1e-08, patience=None)