search

SforAiDl / KD_Lib

A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quantization.
https://kd-lib.readthedocs.io/
MIT License
609 stars 58 forks source link

Fix Test Failures Part 1 #115

Open khizirsiddiqui opened 3 years ago

khizirsiddiqui commented 3 years ago

Fixes few errors raised in tests.

codecov-commenter commented 3 years ago

Codecov Report

Merging #115 (b1b2180) into master (eca9657) will decrease coverage by 0.19%. The diff coverage is 80.00%.

@@            Coverage Diff             @@
##           master     #115      +/-   ##
==========================================
- Coverage   58.23%   58.04%   -0.20%     
==========================================
  Files          60       60              
  Lines        2112     2114       +2     
==========================================
- Hits         1230     1227       -3     
- Misses        882      887       +5
Impacted Files Coverage Δ
KD_Lib/Pruning/lottery_tickets/lottery_tickets.py 41.44% <66.66%> (+0.11%) :arrow_up:
KD_Lib/KD/common/base_class.py 79.35% <100.00%> (ø)
KD_Lib/KD/vision/DML/dml.py 69.36% <100.00%> (ø)
KD_Lib/KD/vision/BANN/BANN.py 58.33% <0.00%> (-6.67%) :arrow_down: