pytorch-pfn-extras
User Guide
API Reference
pytorch_pfn_extras.training.ExtensionsManager
pytorch_pfn_extras.training.IgniteExtensionsManager
pytorch_pfn_extras.training.extension.make_extension
pytorch_pfn_extras.training.extensions.Evaluator
pytorch_pfn_extras.training.extensions.LogReport
pytorch_pfn_extras.training.extensions.MicroAverage
pytorch_pfn_extras.training.extensions.observe_lr
pytorch_pfn_extras.training.extensions.observe_value
pytorch_pfn_extras.training.extensions.ParameterStatistics
pytorch_pfn_extras.training.extensions.PlotReport
pytorch_pfn_extras.training.extensions.PrintReport
pytorch_pfn_extras.training.extensions.ProgressBar
pytorch_pfn_extras.training.extensions.ProfileReport
pytorch_pfn_extras.training.extensions.snapshot
pytorch_pfn_extras.training.extensions.VariableStatisticsPlot
pytorch_pfn_extras.training.triggers.EarlyStoppingTrigger
pytorch_pfn_extras.training.triggers.IntervalTrigger
pytorch_pfn_extras.training.triggers.ManualScheduleTrigger
pytorch_pfn_extras.training.triggers.BestValueTrigger
pytorch_pfn_extras.training.triggers.MaxValueTrigger
pytorch_pfn_extras.training.triggers.MinValueTrigger
pytorch_pfn_extras.training.triggers.OnceTrigger
pytorch_pfn_extras.training.triggers.TimeTrigger
pytorch_pfn_extras.reporting.Reporter
pytorch_pfn_extras.reporting.report
pytorch_pfn_extras.reporting.report_scope
pytorch_pfn_extras.logging.get_logger
pytorch_pfn_extras.profiler.TimeSummary.report
pytorch_pfn_extras.nn.parallel.DistributedDataParallel
pytorch_pfn_extras.utils.checkpoint
pytorch_pfn_extras.nn.LazyLinear
pytorch_pfn_extras.nn.LazyConv1d
pytorch_pfn_extras.nn.LazyConv2d
pytorch_pfn_extras.nn.LazyConv3d
pytorch_pfn_extras.nn.LazyBatchNorm1d
pytorch_pfn_extras.nn.LazyBatchNorm2d
pytorch_pfn_extras.nn.LazyBatchNorm3d
pytorch_pfn_extras.onnx.export
pytorch_pfn_extras.onnx.export_testcase
pytorch_pfn_extras.onnx.annotate
pytorch_pfn_extras.onnx.apply_annotation
pytorch_pfn_extras.onnx.scoped_anchor
pytorch_pfn_extras.onnx.export
pytorch_pfn_extras.onnx.export_testcase
pytorch_pfn_extras.dataset.SharedDataset
pytorch_pfn_extras.dataset.TabularDataset
pytorch_pfn_extras.dataset.ItemNotFoundException
pytorch_pfn_extras.config.Config
pytorch_pfn_extras.config_types.optuna_types
pytorch_pfn_extras.config_types.load_path_with_optuna_types
pytorch_pfn_extras.from_ndarray
pytorch_pfn_extras.as_ndarray
pytorch_pfn_extras.get_xp
pytorch_pfn_extras.as_numpy_dtype
pytorch_pfn_extras.from_numpy_dtype
pytorch_pfn_extras.cuda.stream
pytorch_pfn_extras.cuda.use_torch_mempool_in_cupy
pytorch_pfn_extras.cuda.use_default_mempool_in_cupy
pytorch_pfn_extras.cuda.use_default_mempool_in_cupy
¶
pytorch_pfn_extras.cuda.
use_default_mempool_in_cupy
(
)
¶
Use the default memory pool in CuPy.
pytorch_pfn_extras.cuda.use_torch_mempool_in_cupy