from torchtoolbox.tools import split_weights error

PistonY / torch-toolbox

🛠 Toolbox to extend PyTorch functionalities

BSD 3-Clause "New" or "Revised" License

416 stars 56 forks source link

from torchtoolbox.tools import split_weights error #9

Closed ternaus closed 3 years ago

ternaus commented 3 years ago

In [4]: from torchtoolbox.tools import split_weights                                                                                                                                                                                                                                                 
---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-4-6431db6b00b1> in <module>
----> 1 from torchtoolbox.tools import split_weights

ImportError: cannot import name 'split_weights' from 'torchtoolbox.tools' (/home/vladimir/anaconda3/lib/python3.7/site-packages/torchtoolbox/tools/__init__.py)

PistonY commented 3 years ago

Sorry, I may not update doc, if you want to use no decay bias you should do

from torchtoolbox.tools import no_decay_bias

PistonY commented 3 years ago

The doc has been updated.

ternaus commented 3 years ago

:+1: