We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0f2803d commit 7a0bd09Copy full SHA for 7a0bd09
timm/models/_prune.py
@@ -1,4 +1,5 @@
1
import os
2
+import pkgutil
3
from copy import deepcopy
4
5
from torch import nn as nn
@@ -108,6 +109,5 @@ def adapt_model_from_string(parent_module, model_string):
108
109
110
111
def adapt_model_from_file(parent_module, model_variant):
- adapt_file = os.path.join(os.path.dirname(__file__), '_pruned', model_variant + '.txt')
112
- with open(adapt_file, 'r') as f:
113
- return adapt_model_from_string(parent_module, f.read().strip())
+ adapt_data = pkgutil.get_data(__name__, os.path.join('_pruned', model_variant + '.txt'))
+ return adapt_model_from_string(parent_module, adapt_data.decode('utf-8').strip())
0 commit comments