可以将每个 model 定义为 nn.Sequential 吗？

Question

我正在学习使用 PyTorch。如果有人熟悉 PyTorch，他们能告诉我是否所有模型都可以是 nn.Sequential？

我问是因为某些框架功能只接受定义为 nn.Sequential 的 model 作为输入

Answer 1

是的，但仅在您可以将高度复杂的 model 包装为“nn.Sequential”中的单个步骤的意义上。 如果您想要一个破坏顺序行为的 model 的示例，请查找 ResNet 及其同类。 这些需要在“层”之间传递数据； 然而，即使是那些也可以使用“nn.Sequential”来实现，方法是创建特殊的“nn.Module”类来处理剩余功能，并将这些块堆叠在一起成为顺序。