采用GitHub上的代码
运行类似vgg模型,在cifar10上训练,结果朋友torch与tensorflow速度相当,远远快过keras。
pytorch | tensorflow | keras+tensorflow | |
version | 0.4.0 | 1.8.0 | Keras: 2.1.6 Tensorflow: 1.8.0 |
train time: | 1min 14s | 1min 9s | 1min 51s |
evaluate time: | 378 ms | 9.4 s | 826 ms |
采用GitHub上的代码
运行类似vgg模型,在cifar10上训练,结果朋友torch与tensorflow速度相当,远远快过keras。
pytorch | tensorflow | keras+tensorflow | |
version | 0.4.0 | 1.8.0 | Keras: 2.1.6 Tensorflow: 1.8.0 |
train time: | 1min 14s | 1min 9s | 1min 51s |
evaluate time: | 378 ms | 9.4 s | 826 ms |