Oneflow-Inc / models

Models and examples built with OneFlow
Apache License 2.0
94 stars 37 forks source link

Fix swin dataloader import bug #334

Closed BBuf closed 2 years ago

BBuf commented 2 years ago

/home/zhangxiaoyu/oneflow/python/oneflow/init.py /home/zhangxiaoyu/.local/lib/python3.8/site-packages/flowvision/transforms/transforms.py:745: UserWarning: Argument interpolation should be of type InterpolationMode instead of int. Please, use InterpolationMode enum. warnings.warn( /home/zhangxiaoyu/.local/lib/python3.8/site-packages/torch/init.py OneFlow swin dataloader time: 0.005s (= 0.918s / 200, num_workers=2) PyTorch swin dataloader time: 0.003s (= 0.539s / 200, num_workers=2) Relative speed: 0.587 (= 0.003s / 0.005s)