Skip to content

Commit

Permalink
Exported transforms (#603)
Browse files Browse the repository at this point in the history
Co-authored-by: dblalock <davis@mosaicml.com>
Co-authored-by: hanlint <hanlin@mosaicml.com>
  • Loading branch information
3 people committed Feb 28, 2022
1 parent 4994ceb commit 2c40fcf
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions composer/algorithms/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,10 +40,12 @@ def apply(self, state: State, event: Event, logger: Logger):
from composer.algorithms.algorithm_registry import get_algorithm_registry as get_algorithm_registry
from composer.algorithms.algorithm_registry import list_algorithms as list_algorithms
from composer.algorithms.alibi import Alibi as Alibi
from composer.algorithms.augmix import AugmentAndMixTransform as AugmentAndMixTransform
from composer.algorithms.augmix import AugMix as AugMix
from composer.algorithms.blurpool import BlurPool as BlurPool
from composer.algorithms.channels_last import ChannelsLast as ChannelsLast
from composer.algorithms.colout import ColOut as ColOut
from composer.algorithms.colout import ColOutTransform as ColOutTransform
from composer.algorithms.cutmix import CutMix as CutMix
from composer.algorithms.cutout import CutOut as CutOut
from composer.algorithms.factorize import Factorize as Factorize
Expand Down Expand Up @@ -76,6 +78,7 @@ def apply(self, state: State, event: Event, logger: Logger):
from composer.algorithms.no_op_model import NoOpModel as NoOpModel
from composer.algorithms.progressive_resizing import ProgressiveResizing as ProgressiveResizing
from composer.algorithms.randaugment import RandAugment as RandAugment
from composer.algorithms.randaugment import RandAugmentTransform as RandAugmentTransform
from composer.algorithms.sam import SAM as SAM
from composer.algorithms.scale_schedule import ScaleSchedule as ScaleSchedule
from composer.algorithms.selective_backprop import SelectiveBackprop as SelectiveBackprop
Expand Down

0 comments on commit 2c40fcf

Please sign in to comment.