pytorch-unflow icon indicating copy to clipboard operation
pytorch-unflow copied to clipboard

a reimplementation of UnFlow in PyTorch that matches the official TensorFlow version

Results 1 pytorch-unflow issues
Sort by recently updated
recently updated
newest added

add distributed data parallel in pytorch: add `with dev ` in correlation.py use cupy.RawModule instead of cupy.cuda.compile_with_cache for deprecation in cupy v10 add test_ddp for test