Densenet121 graph and merging issue

gstoica27 / ZipIt

A framework for merging models solving different tasks with different initializations into one multi-task model without any additional training

MIT License

286 stars 25 forks source link

Densenet121 graph and merging issue #29

Open santosh9sanjeev opened 5 months ago

santosh9sanjeev commented 5 months ago

Hi Thank you for the great work.

I am trying to incorporate zipit into DenseNet121 architecture. But I am not able to do it correctly. I am getting errors in handle_conv2d as the shapes of self.merge and module.weight. Could you pls help me with it?

Thanks!