search

kevinzakka / pytorch-goodies

PyTorch Boilerplate For Research
603 stars 71 forks source link

Add code for counting only trainable parameters. #3

Closed ranamihir closed 5 years ago

ranamihir commented 5 years ago

Just a slight modification to count only trainable parameters (useful in transfer learning etc.).

kevinzakka commented 5 years ago

Thanks @ranamihir !