Task

Contents:

• ml.tasks.base
  • num_workers()
  • CumulativeTimer
  • IterationTimer
  • StateTimer
  • DataLoaderConfig
  • DataLoaderConfigs
  • FinishTrainingConfig
  • BaseTaskConfig
  • BaseTask
• ml.tasks.consistency
  • ConsistencyModel
• ml.tasks.datasets.async_iterable
  • add_to_queue()
  • thread_worker()
  • thread_async_iter()
  • AsyncIterableDataset
• ml.tasks.datasets.clippify
  • ClippifyDataset
• ml.tasks.datasets.collate
  • pil_to_tensor()
  • is_named_tuple()
  • pad_sequence()
  • pad_all()
  • collate()
  • collate_non_null()
• ml.tasks.datasets.error_handling
  • get_loc()
  • ErrorHandlingConfig
  • ExceptionSummary
  • ErrorHandlingDataset
  • ErrorHandlingMapDataPipe
  • ErrorHandlingIterableDataset
  • ErrorHandlingIterDataPipe
  • error_handling_dataset()
  • test_exception_summary()
• ml.tasks.datasets.multi_iter
  • DatasetInfo
  • MultiIterDataset
• ml.tasks.datasets.samplers
  • ChunkSampler
• ml.tasks.datasets.streaming
  • StreamingDataset
  • StreamingDatasetNoIndex
• ml.tasks.datasets.transforms
  • square_crop()
  • square_resize_crop()
  • upper_left_crop()
  • normalize()
  • denormalize()
  • random_square_crop()
  • random_square_crop_multi()
  • make_size()
  • make_same_size()
  • SquareResizeCrop
  • UpperLeftCrop
• ml.tasks.datasets.video_file
  • VideoFileDataset
• ml.tasks.diffusion
  • cast_beta_schedule()
  • get_diffusion_beta_schedule()
  • GaussianDiffusion
  • plot_schedules()
• ml.tasks.environments.base
  • Environment
• ml.tasks.environments.utils
  • test_environment()
• ml.tasks.environments.worker
  • clear_queue()
  • cast_worker_mode()
  • BaseEnvironmentWorker
  • SyncEnvironmentWorker
  • AsyncEnvironmentWorker
  • WorkerPool
  • SyncWorkerPool
  • AsyncWorkerPool
  • get_worker_pool()
• ml.tasks.gan.base
  • GenerativeAdversarialNetworkTaskConfig
  • GenerativeAdversarialNetworkTask
• ml.tasks.gan.round_robin
  • GenerativeAdversarialNetworkRoundRobinTaskConfig
  • GenerativeAdversarialNetworkRoundRobinTask
• ml.tasks.losses.audio
  • get_window()
  • stft()
  • spectral_convergence_loss()
  • log_stft_magnitude_loss()
  • stft_magnitude_loss()
  • STFTLoss
  • MultiResolutionSTFTLoss
  • MelLoss
  • MFCCLoss
• ml.tasks.losses.diffusion
  • pseudo_huber_loss()
• ml.tasks.losses.image
  • SSIMLoss
  • ImageGradLoss
  • LPIPS
• ml.tasks.losses.kl
  • kl_div_single_loss()
  • kl_div_pair_loss()
• ml.tasks.losses.loss
  • Modality-Agnostic
  • Audio
  • Image
  • KL Divergence
  • log_cosh_loss()
  • loss_fn()
• ml.tasks.ode
  • cast_solver_type()
  • vanilla_add_fn()
  • BaseODESolver
  • EulerODESolver
  • HeunODESolver
  • RK4ODESolver
  • get_ode_solver()
• ml.tasks.rl.base
  • EnvironmentConfig
  • DatasetConfig
  • ReinforcementLearningTaskConfig
  • ReinforcementLearningTask
• ml.tasks.rl.replay
  • ReplaySamples
  • MultiReplaySamples
  • ReplayDataset
• ml.tasks.sl.base
  • SupervisedLearningTaskConfig
  • SupervisedLearningTask