crumpets.torch package

crumpets.torch.is_cpu_only(val)

checks if val is a value determining cpu-only cuda devices

crumpets.torch.is_single_torch_device(val)

checks if val is a value determining a single cuda device

Submodules

• crumpets.torch.augmentation_cuda module
• crumpets.torch.dataloader module
• crumpets.torch.loss module
• crumpets.torch.metrics module
• crumpets.torch.policy module
• crumpets.torch.randomizer module
• crumpets.torch.shm module
• crumpets.torch.trainer module
• crumpets.torch.utils module