Skip to content
Snippets Groups Projects
  • Neta Zmora's avatar
    1d62f96f
    Fixed a couple of bugs and added tests · 1d62f96f
    Neta Zmora authored
    When removing channels and thinning, the number of filters of the
    next layer was not set correctly.
    
    When loading a model that has already been thinned (e.g when loading
    a model, thinning, saving, loading), don’t crash on wrong tensor sizes.
    
    Cache the thinning recipe in the model when loading from checkpoint.
    Without this, a loaded thin model will lose its recipes when saved to
    checkpoint
    1d62f96f
    History
    Fixed a couple of bugs and added tests
    Neta Zmora authored
    When removing channels and thinning, the number of filters of the
    next layer was not set correctly.
    
    When loading a model that has already been thinned (e.g when loading
    a model, thinning, saving, loading), don’t crash on wrong tensor sizes.
    
    Cache the thinning recipe in the model when loading from checkpoint.
    Without this, a loaded thin model will lose its recipes when saved to
    checkpoint