假设我训练了一个自动编码器。我想冻结编码器的参数以进行训练,因此只有解码器进行训练。assuming it's a single layer called 'encoder'
model.encoder.weights.data.requers_grad = False 或者,我可以只将解码器的参数传递给优化器有什么不同吗?
下面是Datacamp关于使用卷积自动编码器进行分类here的教程。我在教程中了解到,我们只需要将自动编码器的头部(即编码器部分)堆叠到一个完全连接的层上即可进行分类。在堆叠之后,得到的网络(卷积自动编码器)被训练两次。第一种方法是将编码器的权重设置为false,如下所示: for layer in full_