hq-deng / RD4AD

Anomaly Detection via Reverse Distillation from One-Class Embedding
MIT License
174 stars 31 forks source link

为什么编码器设置为评估模式而不是设置为参数冻结 #33

Open rock-lss opened 2 months ago

rock-lss commented 2 months ago

如果要冻结解码器参数为什么用encoder.eval()而不是
for param in encoder.parameters(): param.requires_grad = False