xbtorch.nn.utils

Utility functions for neural network training and testing.

Functions

print_num_unique_values(tensor)
test_classifier(data_loader, model, device)
train_classifier(data_loader, model, ...[, ...])

xbtorch.nn.utils.print_num_unique_values(tensor)

xbtorch.nn.utils.test_classifier(data_loader, model, device, compute_cm=False, log=False)

xbtorch.nn.utils.train_classifier(data_loader, model, criterion, optimizer, epoch, num_epochs=0, device='cpu', lr_decay_rate=1.0, log=False, fast=False)