mapping_networks.trainers.TupleBatchAdapter

class mapping_networks.trainers.TupleBatchAdapter

Unpack (inputs, targets) tuples — the standard PyTorch convention.

Also handles (inputs, targets, *extra) by ignoring extra elements.

__init__()

Methods

__init__()

unpack(batch)

Return ((input_arg1, ...), targets) from a dataloader batch.