AttributeError: module 'oneflow.nn' has no attribute 'DataParallel'

[jackalcooper]

Summary

oneflow.nn doens't have class 'DataParallel'

Code to reproduce bug

model = nn.DataParallel(model)

System Information

• What is your OneFlow installation (pip, source, dockerhub):
• OS:
• OneFlow version (run python3 -m oneflow --doctor):
• Python version:
• CUDA driver version:
• GPU models:
• Other info:

[MARD1NO]

pytorch现在都是推荐用 DDP了，而不是DP