MyGit

v0.10.1

mosaicml/composer

版本发布时间: 2022-10-06 08:17:36

mosaicml/composer最新发布版本:v0.25.0(2024-09-25 04:56:05)

🚀 Composer v0.10.1

Composer v0.10.1 is released! Install via pip:

pip install --upgrade mosaicml==0.10.1

New Features

  1. 𐄷 Weight Standardization

    Weight Standardization reparametrizes convolutional weights such that the fan-in dimensions have zero mean and unit standard deviation. This could slightly improve performance at the expensive of 5% lower throughput. This has been used in several papers to train with smaller batch sizes, with normalization layers besides batch norm, and for transfer learning.

    Using Weight Standardization with the Composer Trainer:

    import composer
    
    # Apply Weight Standardization (when training is initialized)
    weight_std = composer.algorithms.WeightStandardization()
    
    # Train with Weight Standardization
    trainer = composer.trainer.Trainer(
        ...
        algorithms=[weight_std]
    )
    trainer.fit()
    

    Using Weight Standardization with the Composer functional interface:

    import composer
    from torchvision.models import resnet50
    
    my_model = resnet50()
    
    # Apply weight standardization to model
    my_model = composer.functional.weight_standardization(my_model)
    

    Please see the Weight Standardization Method Card for more details.

Bug Fixes

What's Changed

New Contributors

Full Changelog: https://github.com/mosaicml/composer/compare/v0.10.0...v0.10.1

相关地址:原始地址 下载(tar) 下载(zip)

查看:2022-10-06发行的版本