最近在做一个图像分类的小项目,用PyTorch写了个ResNet,跑起来挺顺。但看招聘要求好多都写TensorFlow,又听说TF2.0之后Keras很好上手。现在纠结要不要把主力切到TF上。
这个思路不错,收藏了。
写得挺好,建议补充一些性能数据。
代码生成、辅助开发