From d7d4cd54f00e118767d20a7a70228b704a864f37 Mon Sep 17 00:00:00 2001 From: baemincheon Date: Wed, 17 Jul 2019 15:39:05 +0900 Subject: [PATCH] create sandbox --- get_learning_rate_from_file.py | 17 +++++++++++++++++ learning_rate_schedule_classifier_casia.txt | 7 +++++++ 2 files changed, 24 insertions(+) create mode 100644 get_learning_rate_from_file.py create mode 100644 learning_rate_schedule_classifier_casia.txt diff --git a/get_learning_rate_from_file.py b/get_learning_rate_from_file.py new file mode 100644 index 0000000..960fa49 --- /dev/null +++ b/get_learning_rate_from_file.py @@ -0,0 +1,17 @@ + +def get_learning_rate_from_file(filename, epoch): + with open(filename, 'r') as f: + for line in f.readlines(): + line = line.split('#', 1)[0] + if line: + par = line.strip().split(':') + e = int(par[0]) + if par[1]=='-': + lr = -1 + else: + lr = float(par[1]) + if e <= epoch: + learning_rate = lr + return learning_rate + +print(get_learning_rate_from_file('./learning_rate_schedule_classifier_casia.txt', 1)) \ No newline at end of file diff --git a/learning_rate_schedule_classifier_casia.txt b/learning_rate_schedule_classifier_casia.txt new file mode 100644 index 0000000..29a6c68 --- /dev/null +++ b/learning_rate_schedule_classifier_casia.txt @@ -0,0 +1,7 @@ +# Learning rate schedule +# Maps an epoch number to a learning rate +0: 0.05 +60: 0.005 +80: 0.0005 +91: -1 +