mirrors/deit
1
0
Fork 0
mirror of https://github.com/facebookresearch/deit.git synced 2025-06-03 14:52:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Fix bug] Fix type

Browse Source
This commit is contained in:
brian1009 2023-02-26 08:58:24 +08:00
parent 895eaf3975
commit 60a481f4ce
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sparsity_factory/pruners.py
View File

@ -5,7 +5,7 @@ import numpy as np
ALL_PRUNERS = ['lamp', 'glob', 'unif', 'unifplus', 'erk', 'custom']
def check_valid_pruner(name):
return check_valid_pruner in ALL_PRUNERS
return name in ALL_PRUNERS
def weight_pruner_loader(pruner_string):
"""