bug calling TPL

[charlesmartin14]

This bug can be reproduced by calling the TPL fit on ALBERT layer 17
model_name = f"albert-xxlarge-v2"

details = watcher.analyze(layers=[17], fit='TPL', pl_package='powerlaw', plot=True)

---

AttributeError Traceback (most recent call last)
AttributeError: 'NoneType' object has no attribute 'log10'

The above exception was the direct cause of the following exception:

TypeError Traceback (most recent call last)
in <cell line: 1>()
----> 1 details = watcher.analyze(layers=[17], fit='TPL', pl_package='powerlaw', plot=True)
2 details

2 frames
/usr/local/lib/python3.10/dist-packages/weightwatcher/weightwatcher.py in fit_powerlaw(self, evals, xmin, xmax, plot, layer_name, layer_id, plot_id, sample, sample_size, savedir, savefig, thresh, fix_fingers, finger_thresh, xmin_max, max_fingers, fit_type, pl_package)
3758 plt.title(title)
3759 plt.axvline(x=np.log10(fit.xmin), color='red', label=r'$\lambda_{xmin}$')
-> 3760 plt.axvline(x=np.log10(fit.xmax), color='orange', label=r'$\lambda_{xmax}$')
3761 plt.legend()
3762 if savefig:

TypeError: loop of ufunc does not support argument 0 of type NoneType which has no callable log10 method