export datasets interface

[Flowingsun007]

No description provided.

[jackalcooper]

没看出这两个为什么要放到顶层，这个PR里面也没有用到他们的地方

[github-actions]

CI failed, removing label automerge

[github-actions]

Speed stats:

GPU Name: GeForce GTX 1080 

PyTorch resnet50 time: 148.1ms (= 7402.8ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 126.8ms (= 6340.8ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
Relative speed: 1.17 (= 148.1ms / 126.8ms)

PyTorch resnet50 time: 84.2ms (= 4209.5ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 74.2ms (= 3709.4ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
Relative speed: 1.13 (= 84.2ms / 74.2ms)

PyTorch resnet50 time: 58.3ms (= 2916.3ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 47.5ms (= 2377.3ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
Relative speed: 1.23 (= 58.3ms / 47.5ms)

PyTorch resnet50 time: 45.7ms (= 2284.9ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 41.2ms (= 2062.5ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
Relative speed: 1.11 (= 45.7ms / 41.2ms)

PyTorch resnet50 time: 43.1ms (= 2155.3ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 42.5ms (= 2125.2ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
Relative speed: 1.01 (= 43.1ms / 42.5ms)

[github-actions]

CI failed, removing label automerge

[github-actions]

Speed stats:

GPU Name: GeForce GTX 1080 

PyTorch resnet50 time: 137.4ms (= 6867.6ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 124.1ms (= 6206.6ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
Relative speed: 1.11 (= 137.4ms / 124.1ms)

PyTorch resnet50 time: 81.7ms (= 4086.4ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 72.5ms (= 3623.6ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
Relative speed: 1.13 (= 81.7ms / 72.5ms)

PyTorch resnet50 time: 56.5ms (= 2824.2ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 46.2ms (= 2309.4ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
Relative speed: 1.22 (= 56.5ms / 46.2ms)

PyTorch resnet50 time: 47.3ms (= 2364.9ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 37.3ms (= 1864.8ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
Relative speed: 1.27 (= 47.3ms / 37.3ms)

PyTorch resnet50 time: 43.8ms (= 2188.1ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 37.6ms (= 1878.3ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
Relative speed: 1.16 (= 43.8ms / 37.6ms)