Zj-BinXia / SSL

This project is the official implementation of 'Structured Sparsity Learning for Efficient Video Super-Resolution', CVPR2023
95 stars 6 forks source link

执行self._prune_and_build_new_modelV2()之后,new_model的通道数与原始模型不同 #11

Closed sunyclj closed 11 months ago

sunyclj commented 11 months ago

_prune_and_build_new_modelV2中new_layer根据kept_filter, kept_chl设置,不会包括所有的层,在模型更新时,得到的new_model也就只包含这些层,通道数就与原始模型不对应,在测试时候报错,请问是什么原因呢?

Zj-BinXia commented 11 months ago

你是readme的方法进行测试的吗