• vfdev's avatar
    Normalize, LinearTransformation are scriptable (#2645) · 1b415254
    vfdev authored
    * [WIP] All transforms are now derived from torch.nn.Module
    - Compose, RandomApply, Normalize can be jit scripted
    
    * Fixed flake8
    
    * Updated code and docs
    - added getattr to Lambda and tests
    - updated code and docs of Compose
    - added failing test with append/extend on Composed.transforms
    
    * Fixed flake8
    
    * Updated code, tests and docs
    1b415254
transforms.rst 3.98 KB