Update main.py

wanglouis49 · web-flow · commit bc3dd5acc411 · 2017-11-15T16:24:42.000-05:00
diff --git a/main.py b/main.py
@@ -55,4 +55,4 @@
 
 
 # Save and load the entire model
-torch.save(net.state_dict(), 'models/lenet5_pruned_'+prune_perc+'.pkl')
+torch.save(net.state_dict(), 'models/lenet5_pruned_'+param['prune_perc']+'.pkl')

Original file line number	Diff line number	Diff line change
`@@ -55,4 +55,4 @@`
`55`	`55`
`56`	`56`
`57`	`57`	`# Save and load the entire model`
`58`		`-torch.save(net.state_dict(), 'models/lenet5_pruned_'+prune_perc+'.pkl')`
	`58`	`+torch.save(net.state_dict(), 'models/lenet5_pruned_'+param['prune_perc']+'.pkl')`