Ctrl+K

pytorch-pfn-extras documentation

Site Navigation

  • User Guide
  • API Reference

Site Navigation

  • User Guide
  • API Reference

Section Navigation

  • pytorch_pfn_extras
    • pytorch_pfn_extras.as_ndarray
    • pytorch_pfn_extras.as_numpy_dtype
    • pytorch_pfn_extras.compile
    • pytorch_pfn_extras.from_ndarray
    • pytorch_pfn_extras.from_numpy_dtype
    • pytorch_pfn_extras.get_xp
    • pytorch_pfn_extras.map
    • pytorch_pfn_extras.requires
    • pytorch_pfn_extras.to
    • pytorch_pfn_extras.config
      • pytorch_pfn_extras.config.customize_type
      • pytorch_pfn_extras.config.Any
      • pytorch_pfn_extras.config.Config
      • pytorch_pfn_extras.config.ConfigType
    • pytorch_pfn_extras.config_types
      • pytorch_pfn_extras.config_types.load_path_with_optuna_types
      • pytorch_pfn_extras.config_types.optuna_types
      • pytorch_pfn_extras.config_types.Any
    • pytorch_pfn_extras.cuda
      • pytorch_pfn_extras.cuda.stream
      • pytorch_pfn_extras.cuda.use_default_mempool_in_cupy
      • pytorch_pfn_extras.cuda.use_torch_mempool_in_cupy
    • pytorch_pfn_extras.dataloaders
      • pytorch_pfn_extras.dataloaders.DataLoader
      • pytorch_pfn_extras.dataloaders.dataloader
        • pytorch_pfn_extras.dataloaders.dataloader.default_collate
        • pytorch_pfn_extras.dataloaders.dataloader.default_convert
        • pytorch_pfn_extras.dataloaders.dataloader.get_worker_info
        • pytorch_pfn_extras.dataloaders.dataloader.DataLoader
      • pytorch_pfn_extras.dataloaders.utils
        • pytorch_pfn_extras.dataloaders.utils.Any
        • pytorch_pfn_extras.dataloaders.utils.CollateAsDict
    • pytorch_pfn_extras.dataset
      • pytorch_pfn_extras.dataset.SharedDataset
      • pytorch_pfn_extras.dataset.TabularDataset
      • pytorch_pfn_extras.dataset.ItemNotFoundException
      • pytorch_pfn_extras.dataset.shared_dataset
        • pytorch_pfn_extras.dataset.shared_dataset.Cache
        • pytorch_pfn_extras.dataset.shared_dataset.InfiniteCache
        • pytorch_pfn_extras.dataset.shared_dataset.SharedDataset
        • pytorch_pfn_extras.dataset.shared_dataset.ItemNotFoundException
      • pytorch_pfn_extras.dataset.tabular
        • pytorch_pfn_extras.dataset.tabular.from_data
        • pytorch_pfn_extras.dataset.tabular.DelegateDataset
        • pytorch_pfn_extras.dataset.tabular.delegate_dataset
        • pytorch_pfn_extras.dataset.tabular.from_data
        • pytorch_pfn_extras.dataset.tabular.tabular_dataset
    • pytorch_pfn_extras.distributed
      • pytorch_pfn_extras.distributed.create_distributed_subset_indices
      • pytorch_pfn_extras.distributed.initialize_ompi_environment
      • pytorch_pfn_extras.distributed.DistributedValidationSampler
    • pytorch_pfn_extras.engine
      • pytorch_pfn_extras.engine.NamedTuple
      • pytorch_pfn_extras.engine.cast
      • pytorch_pfn_extras.engine.create_evaluator
      • pytorch_pfn_extras.engine.create_trainer
      • pytorch_pfn_extras.engine.default_transform_model
      • pytorch_pfn_extras.engine.filter_state_objects
      • pytorch_pfn_extras.engine.filter_state_objects_dict
      • pytorch_pfn_extras.engine.Any
      • pytorch_pfn_extras.engine.DistributedEvaluator
      • pytorch_pfn_extras.engine.Evaluator
      • pytorch_pfn_extras.engine.StateObjectProtocol
      • pytorch_pfn_extras.engine.Trainer
    • pytorch_pfn_extras.handler
      • pytorch_pfn_extras.handler.forward
      • pytorch_pfn_extras.handler.torch_autocast
      • pytorch_pfn_extras.handler.update_parameters
      • pytorch_pfn_extras.handler.BaseHandler
      • pytorch_pfn_extras.handler.BaseLogic
      • pytorch_pfn_extras.handler.ClousureLogic
      • pytorch_pfn_extras.handler.CodeBlock
      • pytorch_pfn_extras.handler.CodeBlockLogic
      • pytorch_pfn_extras.handler.Handler
      • pytorch_pfn_extras.handler.Logic
    • pytorch_pfn_extras.logging
      • pytorch_pfn_extras.logging.get_logger
    • pytorch_pfn_extras.nn
      • pytorch_pfn_extras.nn.ensure
      • pytorch_pfn_extras.nn.Ensure
      • pytorch_pfn_extras.nn.ExtendedSequential
      • pytorch_pfn_extras.nn.LazyBatchNorm1d
      • pytorch_pfn_extras.nn.LazyBatchNorm2d
      • pytorch_pfn_extras.nn.LazyBatchNorm3d
      • pytorch_pfn_extras.nn.LazyConv1d
      • pytorch_pfn_extras.nn.LazyConv2d
      • pytorch_pfn_extras.nn.LazyConv3d
      • pytorch_pfn_extras.nn.LazyLinear
      • pytorch_pfn_extras.nn.modules
        • pytorch_pfn_extras.nn.modules.ensure_shape
        • pytorch_pfn_extras.nn.modules.extended_sequential
        • pytorch_pfn_extras.nn.modules.lazy
        • pytorch_pfn_extras.nn.modules.lazy_batchnorm
        • pytorch_pfn_extras.nn.modules.lazy_conv
        • pytorch_pfn_extras.nn.modules.lazy_linear
      • pytorch_pfn_extras.nn.parallel
        • pytorch_pfn_extras.nn.parallel.DistributedDataParallel
        • pytorch_pfn_extras.nn.parallel.distributed
    • pytorch_pfn_extras.onnx
      • pytorch_pfn_extras.onnx.annotate
      • pytorch_pfn_extras.onnx.apply_annotation
      • pytorch_pfn_extras.onnx.as_output
      • pytorch_pfn_extras.onnx.export
      • pytorch_pfn_extras.onnx.export_testcase
      • pytorch_pfn_extras.onnx.grad
      • pytorch_pfn_extras.onnx.is_large_tensor
      • pytorch_pfn_extras.onnx.load_model
      • pytorch_pfn_extras.onnx.no_grad
      • pytorch_pfn_extras.onnx.scoped_anchor
      • pytorch_pfn_extras.onnx.annotate
      • pytorch_pfn_extras.onnx.export_testcase
      • pytorch_pfn_extras.onnx.load
        • pytorch_pfn_extras.onnx.load.load_model
        • pytorch_pfn_extras.onnx.load.Any
        • pytorch_pfn_extras.onnx.load.IO
        • pytorch_pfn_extras.onnx.load.Path
        • pytorch_pfn_extras.onnx.load.Text
      • pytorch_pfn_extras.onnx.strip_large_tensor
        • pytorch_pfn_extras.onnx.strip_large_tensor.is_large_tensor
        • pytorch_pfn_extras.onnx.strip_large_tensor.reduce
      • pytorch_pfn_extras.onnx.symbolic_registry
      • pytorch_pfn_extras.onnx.unstrip_tensor
        • pytorch_pfn_extras.onnx.unstrip_tensor.unstrip
        • pytorch_pfn_extras.onnx.unstrip_tensor.Path
    • pytorch_pfn_extras.ops
      • pytorch_pfn_extras.ops.register
      • pytorch_pfn_extras.ops.OpDesc
      • pytorch_pfn_extras.ops.register
    • pytorch_pfn_extras.profiler
      • pytorch_pfn_extras.profiler.clear_tracer
      • pytorch_pfn_extras.profiler.enable_global_trace
      • pytorch_pfn_extras.profiler.enable_thread_trace
      • pytorch_pfn_extras.profiler.get_time_summary
      • pytorch_pfn_extras.profiler.get_tracer
      • pytorch_pfn_extras.profiler.load_chrome_trace_as_json
      • pytorch_pfn_extras.profiler.record
      • pytorch_pfn_extras.profiler.record_function
      • pytorch_pfn_extras.profiler.record_iterable
      • pytorch_pfn_extras.profiler.ChromeTracer
      • pytorch_pfn_extras.profiler.TimeSummary
      • pytorch_pfn_extras.profiler.TraceableDataset
      • pytorch_pfn_extras.profiler.Tracer
    • pytorch_pfn_extras.reporting
      • pytorch_pfn_extras.reporting.divide_scalar
      • pytorch_pfn_extras.reporting.get_current_reporter
      • pytorch_pfn_extras.reporting.multiply_scalar
      • pytorch_pfn_extras.reporting.overload
      • pytorch_pfn_extras.reporting.report
      • pytorch_pfn_extras.reporting.report_scope
      • pytorch_pfn_extras.reporting.subtract_scalar
      • pytorch_pfn_extras.reporting.Any
      • pytorch_pfn_extras.reporting.DictSummary
      • pytorch_pfn_extras.reporting.Reporter
      • pytorch_pfn_extras.reporting.Summary
    • pytorch_pfn_extras.runtime
      • pytorch_pfn_extras.runtime.BaseRuntime
      • pytorch_pfn_extras.runtime.PyTorchRuntime
    • pytorch_pfn_extras.testing
      • pytorch_pfn_extras.testing.Any
    • pytorch_pfn_extras.torchscript
      • pytorch_pfn_extras.torchscript.find_inplace
      • pytorch_pfn_extras.torchscript.run_jit_pass
      • pytorch_pfn_extras.torchscript.Any
    • pytorch_pfn_extras.training
      • pytorch_pfn_extras.training.make_extension
      • pytorch_pfn_extras.training.AccuracyMetric
      • pytorch_pfn_extras.training.DistributedEvaluator
      • pytorch_pfn_extras.training.Evaluator
      • pytorch_pfn_extras.training.Extension
      • pytorch_pfn_extras.training.ExtensionEntry
      • pytorch_pfn_extras.training.ExtensionsManager
      • pytorch_pfn_extras.training.ExtensionsManagerProtocol
      • pytorch_pfn_extras.training.IgniteExtensionsManager
      • pytorch_pfn_extras.training.StateObjectProtocol
      • pytorch_pfn_extras.training.Trainer
      • pytorch_pfn_extras.training.extension
        • pytorch_pfn_extras.training.extension.make_extension
        • pytorch_pfn_extras.training.extension.Any
        • pytorch_pfn_extras.training.extension.Extension
        • pytorch_pfn_extras.training.extension.ExtensionEntry
        • pytorch_pfn_extras.training.extension.ExtensionsManagerProtocol
      • pytorch_pfn_extras.training.extensions
        • pytorch_pfn_extras.training.extensions.observe_lr
        • pytorch_pfn_extras.training.extensions.observe_value
        • pytorch_pfn_extras.training.extensions.snapshot
        • pytorch_pfn_extras.training.extensions.snapshot_object
        • pytorch_pfn_extras.training.extensions.AverageAccumulate
        • pytorch_pfn_extras.training.extensions.BestValue
        • pytorch_pfn_extras.training.extensions.DistributedEvaluator
        • pytorch_pfn_extras.training.extensions.Evaluator
        • pytorch_pfn_extras.training.extensions.FailOnNonNumber
        • pytorch_pfn_extras.training.extensions.IgniteEvaluator
        • pytorch_pfn_extras.training.extensions.LRScheduler
        • pytorch_pfn_extras.training.extensions.LogReport
        • pytorch_pfn_extras.training.extensions.MaxAccumulate
        • pytorch_pfn_extras.training.extensions.MaxValue
        • pytorch_pfn_extras.training.extensions.MicroAverage
        • pytorch_pfn_extras.training.extensions.MinAccumulate
        • pytorch_pfn_extras.training.extensions.MinValue
        • pytorch_pfn_extras.training.extensions.ParameterStatistics
        • pytorch_pfn_extras.training.extensions.PlotReport
        • pytorch_pfn_extras.training.extensions.PrintReport
        • pytorch_pfn_extras.training.extensions.PrintReportCLI
        • pytorch_pfn_extras.training.extensions.ProfileReport
        • pytorch_pfn_extras.training.extensions.ProgressBar
        • pytorch_pfn_extras.training.extensions.ProgressBarCLI
        • pytorch_pfn_extras.training.extensions.Slack
        • pytorch_pfn_extras.training.extensions.SlackWebhook
        • pytorch_pfn_extras.training.extensions.SnapshotMode
        • pytorch_pfn_extras.training.extensions.StandardDeviationAccumulate
        • pytorch_pfn_extras.training.extensions.TimelineTrace
        • pytorch_pfn_extras.training.extensions.UnbiasedStandardDeviationAccumulate
        • pytorch_pfn_extras.training.extensions.VariableStatisticsPlot
        • pytorch_pfn_extras.training.extensions.accumulate
        • pytorch_pfn_extras.training.extensions.best_value
        • pytorch_pfn_extras.training.extensions.evaluator
        • pytorch_pfn_extras.training.extensions.fail_on_non_number
        • pytorch_pfn_extras.training.extensions.log_report
        • pytorch_pfn_extras.training.extensions.lr_scheduler
        • pytorch_pfn_extras.training.extensions.micro_average
        • pytorch_pfn_extras.training.extensions.parameter_statistics
        • pytorch_pfn_extras.training.extensions.plot_report
        • pytorch_pfn_extras.training.extensions.print_report
        • pytorch_pfn_extras.training.extensions.profile_report
        • pytorch_pfn_extras.training.extensions.progress_bar
        • pytorch_pfn_extras.training.extensions.slack
        • pytorch_pfn_extras.training.extensions.snapshot_writers
        • pytorch_pfn_extras.training.extensions.timeline_trace
        • pytorch_pfn_extras.training.extensions.util
        • pytorch_pfn_extras.training.extensions.value_observation
        • pytorch_pfn_extras.training.extensions.variable_statistics_plot
      • pytorch_pfn_extras.training.manager
      • pytorch_pfn_extras.training.metrics
        • pytorch_pfn_extras.training.metrics.AccuracyMetric
        • pytorch_pfn_extras.training.metrics.Any
      • pytorch_pfn_extras.training.trigger
        • pytorch_pfn_extras.training.trigger.get_trigger
        • pytorch_pfn_extras.training.trigger.IntervalTrigger
        • pytorch_pfn_extras.training.trigger.Trigger
      • pytorch_pfn_extras.training.triggers
        • pytorch_pfn_extras.training.triggers.BestValueTrigger
        • pytorch_pfn_extras.training.triggers.EarlyStoppingTrigger
        • pytorch_pfn_extras.training.triggers.FunctionTrigger
        • pytorch_pfn_extras.training.triggers.IntervalTrigger
        • pytorch_pfn_extras.training.triggers.ManualScheduleTrigger
        • 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.training.triggers.early_stopping_trigger
        • pytorch_pfn_extras.training.triggers.function_trigger
        • pytorch_pfn_extras.training.triggers.interval_trigger
        • pytorch_pfn_extras.training.triggers.manual_schedule_trigger
        • pytorch_pfn_extras.training.triggers.minmax_value_trigger
        • pytorch_pfn_extras.training.triggers.once_trigger
        • pytorch_pfn_extras.training.triggers.time_trigger
    • pytorch_pfn_extras.utils
      • pytorch_pfn_extras.utils.checkpoint
        • pytorch_pfn_extras.utils.checkpoint.checkpoint
        • pytorch_pfn_extras.utils.checkpoint.Any
      • pytorch_pfn_extras.utils.comparer
        • pytorch_pfn_extras.utils.comparer.get_default_comparer
        • pytorch_pfn_extras.utils.comparer.intermediate_value
        • pytorch_pfn_extras.utils.comparer.Any
        • pytorch_pfn_extras.utils.comparer.Comparer
        • pytorch_pfn_extras.utils.comparer.ModelComparer
        • pytorch_pfn_extras.utils.comparer.OutputsComparer
    • pytorch_pfn_extras.writing
      • pytorch_pfn_extras.writing.ProcessQueueWriter
      • pytorch_pfn_extras.writing.ProcessWriter
      • pytorch_pfn_extras.writing.QueueWriter
      • pytorch_pfn_extras.writing.SimpleWriter
      • pytorch_pfn_extras.writing.StandardWriter
      • pytorch_pfn_extras.writing.TensorBoardWriter
      • pytorch_pfn_extras.writing.ThreadQueueWriter
      • pytorch_pfn_extras.writing.ThreadWriter
      • pytorch_pfn_extras.writing.Writer

pytorch_pfn_extras.utils.checkpoint#

Functions

pytorch_pfn_extras.utils.checkpoint.checkpoint(...)

Classes

pytorch_pfn_extras.utils.checkpoint.Any(...)

Special type indicating an unconstrained type.

previous

pytorch_pfn_extras.utils

next

pytorch_pfn_extras.utils.checkpoint.checkpoint
Show Source

© Copyright 2021, Preferred Networks, Inc..

Created using Sphinx 5.3.0.