我想为分类定制最后一层VGG 19架构。这将是更有用的keras或pytorch？

Question

我想为分类问题定制VGG 19架构的最后一层。哪个更有用keras或pytorch？

Answer 1

这在很大程度上取决于你想用它做什么。 虽然Keras提供不同的后端，例如TensorFlow或Theano（这反过来可以为您提供更多的灵活性），并且更好地转移到生产系统， PyTorch肯定也很容易实现。此外，它在（多）GPU系统上提供了出色的扩展性，因为在PyTorch模型中外包计算是微不足道的。我不知道在Keras有多容易（从来没有做过，所以我真的无法判断）。

如果您只想玩其中一个框架，通常归结为个人偏好。我个人更喜欢PyTorch，因为它更多＆＃34; python-esque＆＃34;处理事物，但我知道许多人更喜欢Keras，因为它的布局和文档清晰简洁。

提供更多信息或您的背景信息还可以提高您收到的答案的质量。