• jjsjann123's avatar
    [SyncBatchNorm update] (#285) · ffbb52ba
    jjsjann123 authored
    resolves issue #254
    
    Added input casting for pure python implementation, this supports mismatched
    input and layer dtype.
    ffbb52ba
sync_batchnorm.py 6.23 KB