search

dorimedini / robustness_properties

GLOBIGLOBIGLOBI
0 stars 0 forks source link

Implement pruning in base trainer #87

Closed dorimedini closed 5 years ago

dorimedini commented 5 years ago

Implements #83

@galshachaf something tells me the specific implementation of the pruning process depends on the network architecture more than I took into account... specifically, maybe the _prune method should raise NotImplementedError in BaseTrainer and I should move the implementation to FCTrainer?

galshachaf commented 5 years ago

use get_weighted_layers_indices to get the specific layers that have weights.