rethinking-network-pruning icon indicating copy to clipboard operation
rethinking-network-pruning copied to clipboard

about train() and eval()

Open leonzgtee opened this issue 6 years ago • 4 comments

anyone konws what the line128 model.tarin() and line 154 model.eval() in network-slimming/main.py are meant for. i did not find the defination of these two function,can i just delete them,thanks

leonzgtee avatar Jul 17 '19 02:07 leonzgtee

This is to ensure that the batch norm layer are set accordingly. These two lines should be kept.

Eric-mingjie avatar Jul 17 '19 04:07 Eric-mingjie

thanks for your reply.

leonzgtee avatar Jul 18 '19 07:07 leonzgtee

@leonzgtee did u perform the network slimming, on which architecture how mucha was the performance gain from the main model . did ur accuracy drop ? by how much

abhigoku10 avatar Jul 26 '19 05:07 abhigoku10

@abhigoku10 yes i did the network slimming, in our own training architecture with our own data, map drops about 0.01 from 0.38 to 0.37

leonzgtee avatar Sep 25 '20 09:09 leonzgtee