Hello,
I would like to load a pretrained model, fuse the Conv+BN layers, and save it as a new model.
I have found functions that perform the merging but it seems that I miss the workflow which is pytorch specific, e.g.
- add fusion in the model constructor
- load_state_dict → complains with error because it expects different layer configuration (not fused)
It seems that I should also have modify the model.state_dict but I could not find how this is done.
Thanks in advance,
Panagiotis