Index

_

__bool__() (flame.oneflow.helpers.rank0.FakeObject method) (flame.pytorch.utils.ranking.FakeObject method) __call__() (flame.pytorch.helpers.amp.Amp method) (flame.pytorch.metrics.base_metric.Metric method) (flame.pytorch.metrics.base_metric.MetricList method) (flame.pytorch.metrics.Metric method) (flame.pytorch.metrics.MetricList method) __enter__() (flame.pytorch.trainer.coroutine_scheduler.CoroutineScheduler method) __exit__() (flame.pytorch.trainer.coroutine_scheduler.CoroutineScheduler method) __getattr__() (flame.oneflow.helpers.rank0.FakeObject method) (flame.pytorch.utils.ranking.FakeObject method) __iter__() (flame.oneflow.sampler.DistributedNoPaddingSampler method) (flame.pytorch.sampler.DistributedNoPaddingSampler method) __len__() (flame.oneflow.sampler.DistributedNoPaddingSampler method) (flame.pytorch.sampler.DistributedNoPaddingSampler method) __str__() (flame.core.meters.naive_average_meter.NaiveAverageMeter method) (flame.oneflow.meters.average_meter.AverageMeter method) (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.oneflow.trainer.state_manager.StateManager method) (flame.pytorch.meters.average_meter.AverageMeter method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.AverageMeter method) (flame.pytorch.meters.base_meter.Meter method) (flame.pytorch.meters.EstimatedTimeOfArrival method) (flame.pytorch.meters.LazyAverageMeterDict method) (flame.pytorch.meters.Meter method) (flame.pytorch.meters.time_meter.EstimatedTimeOfArrival method) (flame.pytorch.trainer.state_manager.StateManager method) __version__ (in module flame) _default_compute_fn() (in module flame.pytorch.metrics.base_metric) _DEFAULT_DEVICE (in module flame.oneflow.meters.average_meter) (in module flame.pytorch.meters.average_meter) _default_to_training() (in module flame.oneflow.trainer.state_manager) (in module flame.pytorch.trainer.state_manager) _disable_tqdm() (flame.oneflow.trainer.BaseTrainer method) (flame.oneflow.trainer.trainer.BaseTrainer method) (flame.pytorch.trainer.BaseTrainer method) (flame.pytorch.trainer.trainer.BaseTrainer method) _dist_backend (flame.pytorch.arguments.BaseArgs attribute) _HistoryRecord (in module flame.oneflow.meters.average_meter) (in module flame.pytorch.meters.average_meter) _init_distributed() (in module flame.pytorch.launcher)

_logger (in module flame.core.archiver) (in module flame.core.arguments) (in module flame.core.config) (in module flame.core.config_parser) (in module flame.core.helpers.opencv) (in module flame.core.helpers.tqdm) (in module flame.core.logging) (in module flame.core.operating_system) (in module flame.core.testing) (in module flame.oneflow.dist_run) (in module flame.oneflow.helpers.checkpoint_saver) (in module flame.oneflow.helpers.data) (in module flame.oneflow.helpers.optimizer) (in module flame.oneflow.meters.average_meter) (in module flame.oneflow.trainer.progress_meter) (in module flame.oneflow.trainer.state_manager) (in module flame.oneflow.trainer.trainer) (in module flame.pytorch.arguments) (in module flame.pytorch.distributed) (in module flame.pytorch.engine) (in module flame.pytorch.helpers.amp) (in module flame.pytorch.helpers.checkpoint_saver) (in module flame.pytorch.helpers.cudnn) (in module flame.pytorch.helpers.data) (in module flame.pytorch.helpers.model) (in module flame.pytorch.helpers.optimizer) (in module flame.pytorch.launcher) (in module flame.pytorch.meters.average_meter) (in module flame.pytorch.trainer.progress_meter) (in module flame.pytorch.trainer.state_manager) (in module flame.pytorch.trainer.trainer) (in module flame.pytorch.utils.cudnn) _main_fn() (in module flame.core.main_fn) _parse_dict() (flame.core.config_parser.ConfigParser method) _parse_function() (flame.core.config_parser.ConfigParser method) _parse_list() (flame.core.config_parser.ConfigParser method) _parse_object() (flame.core.config_parser.ConfigParser method) _parse_str() (flame.core.config_parser.ConfigParser method) _SummaryWriter (class in flame.oneflow.trainer.progress_meter) _to_device() (in module flame.oneflow.trainer.trainer) (in module flame.pytorch.trainer.trainer) _to_number() (in module flame.pytorch.metrics.base_metric)

A

add_scalar() (flame.oneflow.trainer.progress_meter._SummaryWriter method) Amp (class in flame.pytorch.helpers.amp) apply (flame.core.arguments.BaseArgs attribute) Args (class in flame.__main__) (class in flame.core.distributed_launcher) (class in flame.oneflow.dist_run) args (in module flame.core.arguments) arrival_time (flame.pytorch.meters.EstimatedTimeOfArrival property) (flame.pytorch.meters.time_meter.EstimatedTimeOfArrival property)

as_dict() (flame.pytorch.trainer_v2.base.Dataclass method) autocast() (flame.pytorch.helpers.amp.Amp method) AverageMeter (class in flame.oneflow.meters.average_meter) (class in flame.pytorch.meters) (class in flame.pytorch.meters.average_meter) avg (flame.core.meters.naive_average_meter.NaiveAverageMeter property) (flame.oneflow.meters.average_meter.AverageMeter property) (flame.pytorch.meters.average_meter.AverageMeter property) (flame.pytorch.meters.AverageMeter property)

B

backward() (flame.pytorch.helpers.amp.Amp method) BaseArgs (class in flame.core.arguments) (class in flame.oneflow.arguments) (class in flame.pytorch.arguments) BaseMeter (class in flame.core.meters.base_meter) BaseTrainer (class in flame.oneflow.trainer) (class in flame.oneflow.trainer.trainer) (class in flame.pytorch.trainer) (class in flame.pytorch.trainer.trainer) (class in flame.pytorch.trainer_v2.trainer)

BatchSizeMetric (class in flame.pytorch.metrics) (class in flame.pytorch.metrics.batch_size_metric)

C

call_with_mapping() (in module flame.pytorch.trainer_v2.context) clip_grad_norm_() (flame.pytorch.helpers.amp.Amp method) config() (flame.core.arguments.BaseArgs method) config_file (flame.core.arguments.BaseArgs attribute) ConfigParser (class in flame.core.config_parser) CONSOLE_DATE_FORMAT (in module flame.core.logging) CONSOLE_FORMAT (in module flame.core.logging) CopyMetric (class in flame.pytorch.metrics) (class in flame.pytorch.metrics.copy_metric) CoroutineScheduler (class in flame.pytorch.trainer.coroutine_scheduler) count (flame.oneflow.meters.average_meter.AverageMeter attribute)

create_console_handler() (in module flame.core.logging) create_data_loader() (in module flame.oneflow.helpers) (in module flame.oneflow.helpers.data) (in module flame.pytorch.helpers) (in module flame.pytorch.helpers.data) create_ddp_model() (in module flame.pytorch.helpers) (in module flame.pytorch.helpers.model) create_file_handler() (in module flame.core.logging) cudnn_benchmark_if_possible() (in module flame.pytorch.helpers) (in module flame.pytorch.helpers.cudnn) cv2_loader() (in module flame.core.helpers.opencv) cv2_worker_init_fn() (in module flame.core.helpers.opencv)

D

Dataclass (class in flame.pytorch.trainer_v2.base) debug (flame.core.arguments.BaseArgs attribute) (flame.oneflow.dist_run.Args attribute) DEFAULT_DEVICE_ID (in module flame.oneflow.arguments) default_main_worker() (in module flame.oneflow.launcher) (in module flame.pytorch.launcher) device (flame.oneflow.arguments.BaseArgs property) (flame.pytorch.arguments.BaseArgs property)

device_id (flame.oneflow.arguments.BaseArgs attribute) dispatch() (flame.core.config_parser.ConfigParser method) dist_backend (flame.pytorch.arguments.BaseArgs property) dist_url (flame.pytorch.arguments.BaseArgs attribute) DistributedNoPaddingSampler (class in flame.oneflow.sampler) (class in flame.pytorch.sampler) do_nothing() (in module flame.oneflow.helpers.rank0) (in module flame.pytorch.utils.ranking) dump_to_json() (in module flame.core.config)

E

elapsed (flame.pytorch.meters.EstimatedTimeOfArrival property) (flame.pytorch.meters.time_meter.EstimatedTimeOfArrival property) elapsed_seconds (flame.pytorch.meters.EstimatedTimeOfArrival property) (flame.pytorch.meters.time_meter.EstimatedTimeOfArrival property) elapsed_time (flame.pytorch.meters.EstimatedTimeOfArrival property) (flame.pytorch.meters.time_meter.EstimatedTimeOfArrival property) emit() (flame.core.logging.TqdmHandler method) empty_tasks() (flame.pytorch.trainer.coroutine_scheduler.CoroutineScheduler method) Engine (class in flame.pytorch.engine) Entry (class in flame.oneflow.trainer.state_manager) (class in flame.pytorch.trainer.state_manager) enumerate() (flame.oneflow.trainer.progress_meter.ProgressMeter method) (flame.pytorch.engine.Engine static method) (flame.pytorch.trainer.progress_meter.ProgressMeter method) epoch (flame.pytorch.engine.EpochState attribute) EPOCH (flame.pytorch.utils.lr_schduler.SchedulePosition attribute) epoch_length (flame.pytorch.engine.IterationState attribute)

epoch_range() (flame.oneflow.trainer.BaseTrainer method) (flame.oneflow.trainer.trainer.BaseTrainer method) (flame.pytorch.trainer.BaseTrainer method) (flame.pytorch.trainer.trainer.BaseTrainer method) EpochState (class in flame.pytorch.engine) EstimatedTimeOfArrival (class in flame.pytorch.meters) (class in flame.pytorch.meters.time_meter) eta (in module flame.pytorch.meters.time_meter) eval() (flame.oneflow.trainer.state.State method) (flame.oneflow.trainer.state_manager.StateManager method) (flame.pytorch.trainer.state.State method) (flame.pytorch.trainer.state_manager.StateManager method) evaluate_file() (in module flame.core.jsonnet) evaluate_snippet() (in module flame.core.jsonnet) every() (flame.pytorch.engine.Engine static method) Experiment (class in flame.pytorch.trainer_v2.expierment) experiment_dir (flame.core.arguments.BaseArgs attribute)

F

FakeObject (class in flame.oneflow.helpers.rank0) (class in flame.pytorch.utils.ranking) FILE_FORMAT (in module flame.core.logging) file_list (in module flame.core.archiver) find_free_port() (in module flame.core.operating_system) flame module flame.__main__ module flame.core module flame.core.archiver module flame.core.arguments module flame.core.ckcfg module flame.core.config module flame.core.config_parser module flame.core.distributed_launcher module flame.core.helpers module flame.core.helpers.multiprocessing module flame.core.helpers.opencv module flame.core.helpers.tqdm module flame.core.jsonnet module flame.core.logging module flame.core.main_fn module flame.core.meters module flame.core.meters.base_meter module flame.core.meters.naive_average_meter module flame.core.meters.tqdm_meter module flame.core.operating_system module flame.core.testing module flame.core.timing module flame.core.zip_importer module flame.oneflow module flame.oneflow.arguments module flame.oneflow.dist_run module flame.oneflow.helpers module flame.oneflow.helpers.checkpoint_saver module flame.oneflow.helpers.data module flame.oneflow.helpers.optimizer module flame.oneflow.helpers.rank0 module flame.oneflow.helpers.tensorboard module flame.oneflow.launcher module flame.oneflow.meters module flame.oneflow.meters.average_meter module flame.oneflow.metrics module flame.oneflow.metrics.functional module flame.oneflow.sampler module flame.oneflow.trainer module flame.oneflow.trainer.progress_meter module flame.oneflow.trainer.state module flame.oneflow.trainer.state_manager module flame.oneflow.trainer.trainer module flame.oneflow.typing_prelude module flame.pytorch module flame.pytorch.arguments module

flame.pytorch.distributed module flame.pytorch.engine module flame.pytorch.experiment module flame.pytorch.helpers module flame.pytorch.helpers.amp module flame.pytorch.helpers.checkpoint_saver module flame.pytorch.helpers.cudnn module flame.pytorch.helpers.data module flame.pytorch.helpers.model module flame.pytorch.helpers.optimizer module flame.pytorch.helpers.tensorboard module flame.pytorch.launcher module flame.pytorch.meters module flame.pytorch.meters.average_meter module flame.pytorch.meters.base_meter module flame.pytorch.meters.smoothed_value module flame.pytorch.meters.time_meter module flame.pytorch.metrics module flame.pytorch.metrics.base_metric module flame.pytorch.metrics.batch_size_metric module flame.pytorch.metrics.copy_metric module flame.pytorch.metrics.functional module flame.pytorch.metrics.loss_metric module flame.pytorch.metrics.metric_dict module flame.pytorch.metrics.topk_accuracy module flame.pytorch.nn module flame.pytorch.nn.momentum_bn module flame.pytorch.sampler module flame.pytorch.trainer module flame.pytorch.trainer.coroutine_scheduler module flame.pytorch.trainer.progress_meter module flame.pytorch.trainer.state module flame.pytorch.trainer.state_manager module flame.pytorch.trainer.trainer module flame.pytorch.trainer_v2 module flame.pytorch.trainer_v2.base module flame.pytorch.trainer_v2.context module flame.pytorch.trainer_v2.expierment module flame.pytorch.trainer_v2.trainer module flame.pytorch.typing_prelude module flame.pytorch.utils module flame.pytorch.utils.attributes module flame.pytorch.utils.cudnn module flame.pytorch.utils.lr_schduler module flame.pytorch.utils.ranking module flame.pytorch.utils.spawn module force (flame.core.arguments.BaseArgs attribute) format_timedelta() (in module flame.core.timing) forward() (flame.pytorch.nn.momentum_bn.MomentumBatchNorm1d method) (flame.pytorch.nn.momentum_bn.MomentumBatchNorm2d method) (flame.pytorch.trainer_v2.base.WeightedLosses method) from_file() (in module flame.core.config) from_snippet() (in module flame.core.config)

G

get() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.oneflow.trainer.progress_meter.ProgressMeter method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method) (flame.pytorch.trainer.progress_meter.ProgressMeter method) get_device_by_backend() (in module flame.pytorch.distributed) get_device_from_module() (in module flame.pytorch.utils.attributes) get_dtype_from_module() (in module flame.pytorch.utils.attributes)

get_first_parameter_or_buffer() (in module flame.pytorch.utils) (in module flame.pytorch.utils.attributes) get_learning_rate_from_optimizer() (in module flame.pytorch.utils) (in module flame.pytorch.utils.attributes) get_rank_safe() (in module flame.pytorch.distributed) get_world_size_safe() (in module flame.pytorch.distributed) git_ls_files() (in module flame.core.archiver) gpu (flame.core.arguments.BaseArgs attribute) (flame.oneflow.dist_run.Args attribute)

I

import_any() (in module flame.core.zip_importer) import_callback() (in module flame.core.jsonnet) import_from_zip() (in module flame.core.zip_importer) init_cpu_process_group() (in module flame.core.testing) init_logging() (in module flame.core.logging) init_process_group_from_file() (flame.pytorch.arguments.BaseArgs method) (in module flame.pytorch.distributed) init_process_group_from_tcp() (flame.pytorch.arguments.BaseArgs method) initialize() (flame.pytorch.trainer_v2.expierment.Experiment method) (flame.pytorch.trainer_v2.trainer.BaseTrainer method) inputs (flame.pytorch.trainer_v2.base.InputsOutputs attribute) (flame.pytorch.trainer_v2.context.MappingConfig attribute)

InputsOutputs (class in flame.pytorch.trainer_v2.base) is_cuda_available() (flame.oneflow.arguments.BaseArgs method) is_dist_available_and_initialized() (in module flame.pytorch.distributed) is_highest() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method) is_lowest() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method) Item (class in flame.pytorch.metrics.metric_dict) iteration (flame.pytorch.engine.IterationState attribute) IterationState (class in flame.pytorch.engine)

J

join() (flame.pytorch.utils.spawn.ProcessContext method)

K

KEY_CALL (in module flame.core.config_parser)

KEY_NAME (in module flame.core.config_parser) KEY_USE (in module flame.core.config_parser)

L

last() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method) launch() (in module flame.core.distributed_launcher) LazyAverageMeterDict (class in flame.oneflow.meters.average_meter) (class in flame.pytorch.meters) (class in flame.pytorch.meters.average_meter) load_state_dict (flame.oneflow.trainer.state_manager.Entry attribute) (flame.pytorch.trainer.state_manager.Entry attribute) load_state_dict() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.oneflow.trainer.state.State method) (flame.oneflow.trainer.state_manager.StateManager method) (flame.pytorch.engine.State method) (flame.pytorch.helpers.amp.Amp method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method) (flame.pytorch.trainer.state.State method) (flame.pytorch.trainer.state_manager.StateManager method)

LoadStateDict (in module flame.oneflow.trainer.state_manager) (in module flame.pytorch.trainer.state_manager) LOCAL_RANK (in module flame.oneflow.arguments) log_code_version() (in module flame.core.logging) log_runtime_env() (in module flame.core.logging) logging_except_hook() (in module flame.core.logging) LossMetric (class in flame.pytorch.metrics) (class in flame.pytorch.metrics.loss_metric) LrScheduler (class in flame.pytorch.utils.lr_schduler) LrSchedulerFn (in module flame.oneflow.typing_prelude) (in module flame.pytorch.typing_prelude) LrSchedulerManager (class in flame.pytorch.utils.lr_schduler)

M

main() (in module flame.__main__) (in module flame.core.ckcfg) (in module flame.oneflow.dist_run) main_fn() (in module flame) (in module flame.core.main_fn) MainWorker (in module flame.oneflow.launcher) (in module flame.pytorch.launcher) make_archive() (in module flame.core.archiver) MappingConfig (class in flame.pytorch.trainer_v2.context) master_addr (flame.oneflow.dist_run.Args attribute) master_port (flame.oneflow.dist_run.Args attribute) max() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method) max_epochs (flame.pytorch.engine.EpochState attribute) max_iterations (flame.pytorch.engine.IterationState attribute) merge_jsonnet() (in module flame.core.config) Meter (class in flame.pytorch.meters) (class in flame.pytorch.meters.base_meter) meter (flame.pytorch.metrics.metric_dict.Item attribute) Metric (class in flame.pytorch.metrics) (class in flame.pytorch.metrics.base_metric) metric (flame.pytorch.metrics.metric_dict.Item attribute) MetricDict (class in flame.pytorch.metrics.metric_dict) MetricList (class in flame.pytorch.metrics) (class in flame.pytorch.metrics.base_metric) metrics (flame.pytorch.engine.EpochState attribute) min() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method) ModelWithLosses (class in flame.pytorch.trainer_v2.base) module flame flame.__main__ flame.core flame.core.archiver flame.core.arguments flame.core.ckcfg flame.core.config flame.core.config_parser flame.core.distributed_launcher flame.core.helpers flame.core.helpers.multiprocessing flame.core.helpers.opencv flame.core.helpers.tqdm flame.core.jsonnet flame.core.logging flame.core.main_fn flame.core.meters flame.core.meters.base_meter flame.core.meters.naive_average_meter flame.core.meters.tqdm_meter flame.core.operating_system flame.core.testing flame.core.timing flame.core.zip_importer flame.oneflow flame.oneflow.arguments flame.oneflow.dist_run flame.oneflow.helpers flame.oneflow.helpers.checkpoint_saver flame.oneflow.helpers.data flame.oneflow.helpers.optimizer flame.oneflow.helpers.rank0 flame.oneflow.helpers.tensorboard flame.oneflow.launcher flame.oneflow.meters flame.oneflow.meters.average_meter flame.oneflow.metrics flame.oneflow.metrics.functional flame.oneflow.sampler flame.oneflow.trainer flame.oneflow.trainer.progress_meter flame.oneflow.trainer.state flame.oneflow.trainer.state_manager flame.oneflow.trainer.trainer flame.oneflow.typing_prelude flame.pytorch flame.pytorch.arguments flame.pytorch.distributed flame.pytorch.engine flame.pytorch.experiment flame.pytorch.helpers flame.pytorch.helpers.amp flame.pytorch.helpers.checkpoint_saver flame.pytorch.helpers.cudnn flame.pytorch.helpers.data flame.pytorch.helpers.model flame.pytorch.helpers.optimizer flame.pytorch.helpers.tensorboard flame.pytorch.launcher flame.pytorch.meters flame.pytorch.meters.average_meter flame.pytorch.meters.base_meter flame.pytorch.meters.smoothed_value flame.pytorch.meters.time_meter flame.pytorch.metrics flame.pytorch.metrics.base_metric flame.pytorch.metrics.batch_size_metric flame.pytorch.metrics.copy_metric flame.pytorch.metrics.functional flame.pytorch.metrics.loss_metric flame.pytorch.metrics.metric_dict flame.pytorch.metrics.topk_accuracy flame.pytorch.nn flame.pytorch.nn.momentum_bn flame.pytorch.sampler flame.pytorch.trainer flame.pytorch.trainer.coroutine_scheduler flame.pytorch.trainer.progress_meter flame.pytorch.trainer.state flame.pytorch.trainer.state_manager flame.pytorch.trainer.trainer flame.pytorch.trainer_v2 flame.pytorch.trainer_v2.base flame.pytorch.trainer_v2.context flame.pytorch.trainer_v2.expierment flame.pytorch.trainer_v2.trainer flame.pytorch.typing_prelude flame.pytorch.utils flame.pytorch.utils.attributes flame.pytorch.utils.cudnn flame.pytorch.utils.lr_schduler flame.pytorch.utils.ranking flame.pytorch.utils.spawn

module (flame.oneflow.dist_run.Args attribute) module_name (flame.oneflow.trainer.BaseTrainer property) (flame.oneflow.trainer.trainer.BaseTrainer property) (flame.pytorch.trainer.BaseTrainer property) (flame.pytorch.trainer.trainer.BaseTrainer property) momentum_cosine_decay() (flame.pytorch.nn.momentum_bn.MomentumBatchNorm1d method) (flame.pytorch.nn.momentum_bn.MomentumBatchNorm2d method) MomentumBatchNorm1d (class in flame.pytorch.nn.momentum_bn) MomentumBatchNorm2d (class in flame.pytorch.nn.momentum_bn)

N

NaiveAverageMeter (class in flame.core.meters.naive_average_meter) name (flame.oneflow.trainer.BaseTrainer property) (flame.oneflow.trainer.trainer.BaseTrainer property) (flame.pytorch.trainer.BaseTrainer property) (flame.pytorch.trainer.trainer.BaseTrainer property) named_meters() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method)

no_python (flame.core.distributed_launcher.Args attribute) (flame.oneflow.dist_run.Args attribute) no_tqdm (flame.core.arguments.BaseArgs attribute) num_valid_samples() (in module flame.oneflow.sampler) (in module flame.pytorch.distributed) num_valid_samples_from_data_loader() (in module flame.pytorch.distributed) (in module flame.pytorch.helpers)

O

OptimizerFn (in module flame.oneflow.typing_prelude) (in module flame.pytorch.typing_prelude)

outputs (flame.pytorch.trainer_v2.base.InputsOutputs attribute) (flame.pytorch.trainer_v2.context.MappingConfig attribute)

P

ParsableConfig (in module flame.core.config_parser) parse() (flame.core.config_parser.ConfigParser method) parse_config() (in module flame.core.config) parse_gpu_list() (in module flame.core.arguments) parse_root_config() (flame.core.config_parser.ConfigParser method) PATCH_FILE (in module flame.core.config) pids() (flame.pytorch.utils.spawn.ProcessContext method) PREFIX_IMPORT (in module flame.core.config_parser)

PREFIX_PLACEHOLDER (in module flame.core.config_parser) print_freq (flame.core.arguments.BaseArgs attribute) ProcessContext (class in flame.pytorch.utils.spawn) progress_meter() (flame.oneflow.trainer.BaseTrainer method) (flame.oneflow.trainer.trainer.BaseTrainer method) (flame.pytorch.trainer.BaseTrainer method) (flame.pytorch.trainer.trainer.BaseTrainer method) ProgressMeter (class in flame.oneflow.trainer.progress_meter) (class in flame.pytorch.trainer.progress_meter)

R

R (in module flame.pytorch.trainer_v2.context) rank0() (in module flame.oneflow.helpers.rank0) (in module flame.pytorch.helpers) (in module flame.pytorch.utils) (in module flame.pytorch.utils.ranking) Rank0SummaryWriter (class in flame.oneflow.helpers.tensorboard) (class in flame.pytorch.helpers.tensorboard) rank_start (flame.oneflow.dist_run.Args attribute) (flame.pytorch.arguments.BaseArgs attribute) rate (flame.pytorch.meters.EstimatedTimeOfArrival property) (flame.pytorch.meters.time_meter.EstimatedTimeOfArrival property) record() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method) redirect_stdout_and_stderr (flame.oneflow.dist_run.Args attribute) reduce_numbers() (in module flame.pytorch.distributed) register() (flame.oneflow.trainer.state_manager.StateManager method) (flame.pytorch.trainer.state_manager.StateManager method) register_grad_scaler() (flame.oneflow.trainer.state_manager.StateManager method) (flame.pytorch.trainer.state_manager.StateManager method) register_lr_scheduler() (flame.oneflow.trainer.state_manager.StateManager method) (flame.pytorch.trainer.state_manager.StateManager method) register_model() (flame.oneflow.trainer.state_manager.StateManager method) (flame.pytorch.trainer.state_manager.StateManager method) register_optimizer() (flame.oneflow.trainer.state_manager.StateManager method) (flame.pytorch.trainer.state_manager.StateManager method) remaining (flame.pytorch.meters.EstimatedTimeOfArrival property) (flame.pytorch.meters.time_meter.EstimatedTimeOfArrival property) remaining_seconds (flame.pytorch.meters.EstimatedTimeOfArrival property) (flame.pytorch.meters.time_meter.EstimatedTimeOfArrival property)

remaining_time (flame.pytorch.meters.EstimatedTimeOfArrival property) (flame.pytorch.meters.time_meter.EstimatedTimeOfArrival property) remap_inputs() (in module flame.pytorch.trainer_v2.context) remap_outputs() (in module flame.pytorch.trainer_v2.context) remove_all_handlers() (in module flame.core.logging) require() (in module flame.core.config_parser) reset() (flame.core.meters.base_meter.BaseMeter method) (flame.core.meters.naive_average_meter.NaiveAverageMeter method) (flame.oneflow.meters.average_meter.AverageMeter method) (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.average_meter.AverageMeter method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.AverageMeter method) (flame.pytorch.meters.base_meter.Meter method) (flame.pytorch.meters.EstimatedTimeOfArrival method) (flame.pytorch.meters.LazyAverageMeterDict method) (flame.pytorch.meters.Meter method) (flame.pytorch.meters.time_meter.EstimatedTimeOfArrival method) resume (flame.core.arguments.BaseArgs attribute) resume() (flame.oneflow.trainer.state_manager.StateManager method) (flame.pytorch.trainer.state_manager.StateManager method) run() (flame.oneflow.trainer.BaseTrainer method) (flame.oneflow.trainer.trainer.BaseTrainer method) (flame.pytorch.engine.Engine method) (flame.pytorch.trainer.BaseTrainer method) (flame.pytorch.trainer.coroutine_scheduler.CoroutineScheduler method) (flame.pytorch.trainer.trainer.BaseTrainer method) (flame.pytorch.trainer_v2.trainer.BaseTrainer method) run_distributed() (in module flame.oneflow.launcher) (in module flame.pytorch.launcher) run_once() (flame.pytorch.trainer_v2.trainer.BaseTrainer method)

S

safe_delete() (in module flame.oneflow.helpers.checkpoint_saver) (in module flame.pytorch.helpers.checkpoint_saver) save_checkpoint() (in module flame.oneflow.helpers) (in module flame.oneflow.helpers.checkpoint_saver) (in module flame.pytorch.helpers) (in module flame.pytorch.helpers.checkpoint_saver) save_command() (flame.core.arguments.BaseArgs method) save_config() (flame.core.arguments.BaseArgs method) save_history() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method) scale() (flame.pytorch.helpers.amp.Amp method) scale_lr_linearly() (in module flame.oneflow.helpers) (in module flame.oneflow.helpers.optimizer) (in module flame.pytorch.helpers) (in module flame.pytorch.helpers.optimizer) SchedulePosition (class in flame.pytorch.utils.lr_schduler) set_cudnn_benchmark() (in module flame.pytorch.utils) (in module flame.pytorch.utils.cudnn) set_logging_except_hook() (in module flame.core.logging) set_sampler_epoch() (flame.oneflow.trainer.BaseTrainer method) (flame.oneflow.trainer.trainer.BaseTrainer method) (flame.pytorch.trainer.BaseTrainer method) (flame.pytorch.trainer.trainer.BaseTrainer method) spawn() (in module flame.pytorch.utils.spawn) start_processes() (in module flame.pytorch.utils.spawn) State (class in flame.oneflow.trainer.state) (class in flame.pytorch.engine) (class in flame.pytorch.trainer.state)

state_dict() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.oneflow.trainer.state.State method) (flame.oneflow.trainer.state_manager.StateManager method) (flame.pytorch.engine.State method) (flame.pytorch.helpers.amp.Amp method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method) (flame.pytorch.trainer.state.State method) (flame.pytorch.trainer.state_manager.StateManager method) StateManager (class in flame.oneflow.trainer.state_manager) (class in flame.pytorch.trainer.state_manager) STEP (flame.pytorch.utils.lr_schduler.SchedulePosition attribute) step() (flame.pytorch.engine.Engine method) (flame.pytorch.helpers.amp.Amp method) (flame.pytorch.utils.lr_schduler.LrScheduler method) sum (flame.oneflow.meters.average_meter.AverageMeter attribute) sync() (flame.core.meters.base_meter.BaseMeter method) (flame.oneflow.meters.average_meter.AverageMeter method) (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.average_meter.AverageMeter method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.AverageMeter method) (flame.pytorch.meters.base_meter.Meter method) (flame.pytorch.meters.LazyAverageMeterDict method) (flame.pytorch.meters.Meter method)

T

T (flame.oneflow.trainer.BaseTrainer attribute) (flame.oneflow.trainer.trainer.BaseTrainer attribute) (flame.pytorch.trainer.BaseTrainer attribute) (flame.pytorch.trainer.trainer.BaseTrainer attribute) (in module flame.core.main_fn) (in module flame.oneflow.helpers.rank0) (in module flame.oneflow.trainer.trainer) (in module flame.pytorch.trainer.trainer) (in module flame.pytorch.utils.ranking) temp_dir (flame.core.arguments.BaseArgs attribute) TensorOrIterableTensors (in module flame.pytorch.helpers.amp) test() (flame.oneflow.trainer.BaseTrainer method) (flame.oneflow.trainer.trainer.BaseTrainer method) (flame.pytorch.trainer.BaseTrainer method) (flame.pytorch.trainer.trainer.BaseTrainer method) test_state (flame.pytorch.engine.EpochState attribute) to_device() (flame.oneflow.trainer.BaseTrainer method) (flame.oneflow.trainer.trainer.BaseTrainer method) (flame.pytorch.trainer.BaseTrainer method) (flame.pytorch.trainer.trainer.BaseTrainer method) to_state_dict (flame.oneflow.trainer.state_manager.Entry attribute) (flame.pytorch.trainer.state_manager.Entry attribute) to_str() (flame.oneflow.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.average_meter.LazyAverageMeterDict method) (flame.pytorch.meters.LazyAverageMeterDict method) to_training (flame.oneflow.trainer.state_manager.Entry attribute) (flame.pytorch.trainer.state_manager.Entry attribute)

topk_accuracy() (in module flame.oneflow.metrics.functional) (in module flame.pytorch.metrics.functional) TopkAccuracy (class in flame.pytorch.metrics) (class in flame.pytorch.metrics.topk_accuracy) ToStateDict (in module flame.oneflow.trainer.state_manager) (in module flame.pytorch.trainer.state_manager) ToTraining (in module flame.oneflow.trainer.state_manager) (in module flame.pytorch.trainer.state_manager) tqdm_get_rate() (in module flame.core.helpers.tqdm) tqdm_get_total_time() (in module flame.core.helpers.tqdm) TqdmHandler (class in flame.core.logging) TqdmMeter (class in flame.core.meters.tqdm_meter) train() (flame.oneflow.trainer.BaseTrainer method) (flame.oneflow.trainer.state.State method) (flame.oneflow.trainer.state_manager.StateManager method) (flame.oneflow.trainer.trainer.BaseTrainer method) (flame.pytorch.trainer.BaseTrainer method) (flame.pytorch.trainer.state.State method) (flame.pytorch.trainer.state_manager.StateManager method) (flame.pytorch.trainer.trainer.BaseTrainer method) train_state (flame.pytorch.engine.EpochState attribute) training_script (flame.oneflow.dist_run.Args attribute) training_script_args (flame.oneflow.dist_run.Args attribute) try_cuda_set_device() (flame.pytorch.arguments.BaseArgs method) try_make_experiment_dir() (flame.core.arguments.BaseArgs method) try_path() (in module flame.core.jsonnet)

U

ulimit_n_max() (in module flame.core.operating_system) unscale_() (flame.pytorch.helpers.amp.Amp method) update() (flame.core.meters.naive_average_meter.NaiveAverageMeter method) (flame.core.meters.tqdm_meter.TqdmMeter method) (flame.oneflow.meters.average_meter.AverageMeter method) (flame.pytorch.helpers.amp.Amp method) (flame.pytorch.meters.average_meter.AverageMeter method) (flame.pytorch.meters.AverageMeter method) (flame.pytorch.meters.EstimatedTimeOfArrival method) (flame.pytorch.meters.time_meter.EstimatedTimeOfArrival method)

update_batch_size() (flame.oneflow.trainer.progress_meter.ProgressMeter method) (flame.pytorch.trainer.progress_meter.ProgressMeter method)

V

val (flame.oneflow.meters.average_meter.AverageMeter attribute) val_state (flame.pytorch.engine.EpochState attribute) validate() (flame.oneflow.trainer.BaseTrainer method) (flame.oneflow.trainer.trainer.BaseTrainer method) (flame.pytorch.trainer.BaseTrainer method) (flame.pytorch.trainer.trainer.BaseTrainer method)

W

WeightedLosses (class in flame.pytorch.trainer_v2.base) world_size (flame.oneflow.dist_run.Args attribute) (flame.pytorch.arguments.BaseArgs attribute)

write_summary() (flame.oneflow.trainer.progress_meter.ProgressMeter method) (flame.pytorch.trainer.progress_meter.ProgressMeter method)

Z

zip_files() (in module flame.core.archiver)