Implement Lottery Ticket Hypothesis support

ben-hawks / pytorch-jet-classify

Pytorch implementation of the HLS4ML 3 Layer Jet Tagging model, including a standard Pytorch (float) and a Quantized (via Xilinx's Brevitas library) implementation. designed for use with the HLS4ML LHC Jet dataset (100 particles): https://zenodo.org/record/3602254

8 stars 8 forks source link

Implement Lottery Ticket Hypothesis support #7

Closed ben-hawks closed 3 years ago

ben-hawks commented 4 years ago

Support training with the lottery ticket hypothesis pruning method

ben-hawks commented 3 years ago

Added in e1c9aed758f2e4a50d3399d9c65183ac0e7debf0