Xin Yao
Xin Yao
@guanfuchen 1. The official implementation of MoCo also uses PIL GaussianBlur, however I haven't tried it because it accepts a single parameter `sigma`. I kept the implementation as close to...
@youweiliang See #7
> Have you done any regression test? Not yet. It seems I don't have the authentication to trigger regression tests, and I failed to run it locally. > How is...
The decorator should be necessary for custom autograd modules. One alternative might be [`_cast_if_autocast_enabled`](https://github.com/NVIDIA/apex/blob/2e025ab5aa086b003165ad8bf30fa4999c8f6e92/apex/_autocast_utils.py#L19-L23) from Apex. An example is [here](https://github.com/NVIDIA/apex/blob/2e025ab5aa086b003165ad8bf30fa4999c8f6e92/apex/normalization/fused_layer_norm.py#L174-L177). It's also helpful to make custom autograd modules work with...
Generally LGTM.