Commits · main · llvm / yapt

13 Mar, 2022 3 commits
- Add deepcopy / use implicit device · ae767db6
  Yifan Zhao authored Mar 13, 2022
  
  ae767db6
- Better configuration dumping · b3b5a2a2
  Yifan Zhao authored Mar 13, 2022
  
  b3b5a2a2
- Fixed train/eval mode confusion · 3e100855
  Yifan Zhao authored Mar 13, 2022
  
  3e100855
09 Mar, 2022 1 commit
- Fixed channel/filter confusion: channel -> filter · ca17f31b
  Yifan Zhao authored Mar 09, 2022
  
  ca17f31b
28 Feb, 2022 3 commits
- Fixed pytorch-lightning version requirements · f96bf158
  Yifan Zhao authored Feb 28, 2022
  
  f96bf158
- Fixed an issue with example_input_array · 14073b24
  Yifan Zhao authored Feb 28, 2022
  
  14073b24
- Removed unused files, outdated req, and updated readme.md · 12195624
  Yifan Zhao authored Feb 28, 2022
  
  12195624
22 Feb, 2022 3 commits
- Some reformatting · 04ca0876
  Yifan Zhao authored Feb 22, 2022
  
  04ca0876
- Improved StructuredPruningCallback · 965edf14
  Yifan Zhao authored Feb 22, 2022
  
  965edf14
- Added EagleEye pruning · 449a6ad7
  Yifan Zhao authored Feb 21, 2022
  
  449a6ad7
20 Feb, 2022 6 commits
- Reorganized structure pruning directory · 1eadcfe1
  Yifan Zhao authored Feb 20, 2022
  
  1eadcfe1
- Added functionality to prepare for more pruning methods · dd2a6805
  Yifan Zhao authored Feb 20, 2022
  
  dd2a6805
- Added test for all image classification net architectures in torchvision · aa6b10f8
  Yifan Zhao authored Feb 20, 2022
  
  aa6b10f8
- Organize supported operators better · f246c9c0
  Yifan Zhao authored Feb 20, 2022
```
Added more operators; find functions by the function object, not name
```
  f246c9c0
- Fixed failed tests due to pytorch version change · b7edf68f
  Yifan Zhao authored Feb 19, 2022
  
  b7edf68f
- Improved CFG visual and type annotation · a482c6d8
  Yifan Zhao authored Feb 19, 2022
  
  a482c6d8
19 Feb, 2022 1 commit
- Fixed issues on last pruning epoch · 7cce1bb1
  Yifan Zhao authored Feb 19, 2022
  
  7cce1bb1
03 Feb, 2022 4 commits
- Slightly extended DefaultShapeHandler · ee0ebb15
  Yifan Zhao authored Feb 03, 2022
  
  ee0ebb15
- Supports loading directly from pruned checkpoint · 7e4a8489
  Yifan Zhao authored Feb 03, 2022
  
  7e4a8489
- Added lr rewinding and checkpoint reloading · cad560e1
  Yifan Zhao authored Feb 03, 2022
  
  cad560e1
- Improved checkpointing and accuracy logging · f1f2bcd8
  Yifan Zhao authored Feb 03, 2022
  
  f1f2bcd8
16 Dec, 2021 1 commit
- _run_pruning -> prune_module_ (so it can be used directly) · 155f8698
  Yifan Zhao authored Dec 16, 2021
  
  155f8698
14 Oct, 2021 1 commit
- Fixed a problem about onnx_output_names · f24ac877
  Yifan Zhao authored Oct 14, 2021
  
  f24ac877
09 Oct, 2021 1 commit
- Fixed misuse of pl_module.result · 1c597c44
  Yifan Zhao authored Oct 09, 2021
  
  1c597c44
08 Oct, 2021 1 commit
- Implemented pruning on ConvTranspose2D · a1d87386
  Yifan Zhao authored Oct 08, 2021
  
  a1d87386
07 Oct, 2021 1 commit
- Fixed pruning not going all the way till last channel · 36d2feeb
  Yifan Zhao authored Oct 07, 2021
  
  36d2feeb
13 Sep, 2021 2 commits
- Fixed sample_input on custom-collated dataloader · 73219e1f
  Yifan Zhao authored Sep 13, 2021
  
  73219e1f
- Fixed last prune step not saved · e79e76a9
  Yifan Zhao authored Sep 13, 2021
  
  e79e76a9
14 Aug, 2021 2 commits
- Added capability to prune the conv layer before linear layer · d281013a
  Yifan Zhao authored Aug 14, 2021
  
  d281013a
- Fixed a bug when dealing with layers that prunes 0 channels · 3cbb828c
  Yifan Zhao authored Aug 14, 2021
  
  3cbb828c
13 Aug, 2021 2 commits
- Added switch on ONNX model export · 30786c22
  Yifan Zhao authored Aug 13, 2021
  
  30786c22
- Remove a portion of filters from every layer · 0871c328
  Yifan Zhao authored Aug 13, 2021
  
  0871c328
12 Aug, 2021 8 commits
- Fixed a problem · ce23699d
  Yifan Zhao authored Aug 12, 2021
  
  ce23699d
- Improved pruning logging · 31c0d178
  Yifan Zhao authored Aug 12, 2021
  
  31c0d178
- Added protection ratio for layer to prevent all channels being pruned · 3b55c7a3
  Yifan Zhao authored Aug 12, 2021
  
  3b55c7a3
- Reformatted files · 985d2256
  Yifan Zhao authored Aug 12, 2021
  
  985d2256
- Added test for callback (end to end) and fixed some problems · 618b3b5c
  Yifan Zhao authored Aug 12, 2021
  
  618b3b5c
- Fixed and improved setup.cfg · 7b40c27a
  Yifan Zhao authored Aug 12, 2021
  
  7b40c27a
- This only works with lightning 1.3.x · c4ba3811
  Yifan Zhao authored Aug 12, 2021
  
  c4ba3811
- Updated env requirement · b61bf678
  Yifan Zhao authored Aug 12, 2021
  
  b61bf678