torch.utils.cpp_extension — PyTorch 1.10 documentation
pytorch.org › docs › stabletorch.utils.cpp_extension.CUDAExtension(name, sources, *args, **kwargs) [source] Creates a setuptools.Extension for CUDA/C++. Convenience method that creates a setuptools.Extension with the bare minimum (but often sufficient) arguments to build a CUDA/C++ extension. This includes the CUDA include path, library path and runtime library.
torch.utils.data — PyTorch 1.10 documentation
https://pytorch.org/docs/stable/datatorch.utils.data. At the heart of PyTorch data loading utility is the torch.utils.data.DataLoader class. It represents a Python iterable over a dataset, with support for. map-style and iterable-style datasets, customizing data loading order, automatic batching, single- and multi-process data loading, automatic memory pinning.
pytorch-utils · PyPI
pypi.org › project › pytorch-utilsJan 29, 2019 · pip install pytorch-utils. Copy PIP instructions. Latest version. Released: Jan 29, 2019. Utilities for training and building models in pytorch. Project description. Project details. Release history. Download files.
torch.utils.tensorboard — PyTorch 1.10 documentation
https://pytorch.org/docs/stable/tensorboardclass torch.utils.tensorboard.writer. SummaryWriter (log_dir = None, comment = '', purge_step = None, max_queue = 10, flush_secs = 120, filename_suffix = '') [source] ¶. Writes entries directly to event files in the log_dir to be consumed by TensorBoard. The SummaryWriter class provides a high-level API to create an event file in a given directory and add summaries and events to it.