kylefang/fastai
1
0
Fork 0
mirror of https://github.com/zhigang1992/fastai.git synced 2026-04-24 05:14:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #430 from JanLauGe/master

Browse Source 
modify LR_Finder.plot to use n_skip_end=0
This commit is contained in:
Jeremy Howard
2018-05-02 14:42:02 -07:00
committed by GitHub
parent 784d3f64cf 25169975f4
commit fee6824e73
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fastai/sgdr.py
View File

@@ -158,7 +158,7 @@ class LR_Finder(LR_Updater):
def plot(self, n_skip=10, n_skip_end=5):
plt.ylabel("loss")
plt.xlabel("learning rate (log scale)")
plt.plot(self.lrs[n_skip:-n_skip_end], self.losses[n_skip:-n_skip_end])
plt.plot(self.lrs[n_skip:-(n_skip_end+1)], self.losses[n_skip:-(n_skip_end+1)])
plt.xscale('log')
class LR_Finder2(LR_Finder):