def _train(cls, **kwargs): return train(**kwargs)
def _train(cls, train_loader, model, criterion, optimizer, writer, epoch, **kwargs): return train.train(train_loader, model, criterion, optimizer, writer, epoch, **kwargs)