Create one_cycle() function (#1836)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-01-04 15:49:08 -08:00
1 parent 7dddb1d928
commit 0e341c5660
3 files changed
+9 -3

No files matched your search

+1
View File
@@ -190,6 +190,7 @@ def plot_lr_scheduler(optimizer, scheduler, epochs=300, save_dir=''):
plt.xlim(0, epochs)
plt.ylim(0)
plt.savefig(Path(save_dir) / 'LR.png', dpi=200)
plt.close()
def plot_test_txt(): # from utils.plots import *; plot_test()