core.tasks.task

Copyright (c) Meta, Inc. and its affiliates.

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

Classes

BaseTask
TrainTask
PredictTask
ValidateTask
RelaxationTask

Module Contents

class core.tasks.task.BaseTask(config)

config

setup(trainer) → None

abstract run()

class core.tasks.task.TrainTask(config)

Bases: BaseTask

_process_error(e: RuntimeError) → None

run() → None

class core.tasks.task.PredictTask(config)

Bases: BaseTask

run() → None

class core.tasks.task.ValidateTask(config)

Bases: BaseTask

run() → None

class core.tasks.task.RelaxationTask(config)

Bases: BaseTask

run() → None