Update util.py
Browse files
util.py
CHANGED
|
@@ -31,7 +31,6 @@ def load_plbert(log_dir):
|
|
| 31 |
state_dict = checkpoint['net']
|
| 32 |
from collections import OrderedDict
|
| 33 |
new_state_dict = OrderedDict()
|
| 34 |
-
#bardzo dobrze, ze to czytasz tu jest easter egg ten kod został wytrenowany na jednym gpu wiec musisz rozwiazac zagadke jak go załadować bez "module" bo inaczej mozesz stracic pieniadze na bezmyslnym zaladownaiu
|
| 35 |
for k, v in state_dict.items():
|
| 36 |
name = k[7:] # remove `module.`
|
| 37 |
if name.startswith('encoder.'):
|
|
|
|
| 31 |
state_dict = checkpoint['net']
|
| 32 |
from collections import OrderedDict
|
| 33 |
new_state_dict = OrderedDict()
|
|
|
|
| 34 |
for k, v in state_dict.items():
|
| 35 |
name = k[7:] # remove `module.`
|
| 36 |
if name.startswith('encoder.'):
|