-
Benjamin Vandersmissen authored
Added a function to load weights and pruning masks from the numpy saved files. This is in case that we want to do further pruning. Small fix for VGG, to prevent pruning the output layer.
009fd410
Added a function to load weights and pruning masks from the numpy saved files. This is in case that we want to do further pruning. Small fix for VGG, to prevent pruning the output layer.