diff options
| author | JC-Array <44535867+JC-Array@users.noreply.github.com> | 2022-10-11 22:33:15 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-11 22:33:15 +0000 |
| commit | 963d98639673098fa8df975dd380f1ef56fff3b5 (patch) | |
| tree | 21d41f53af03ce2b21de7947fc216784fb2f2364 /modules/textual_inversion/learn_schedule.py | |
| parent | ff4ef13dd591ec52f196f344f47537695df95364 (diff) | |
| parent | 6be32b31d181e42c639dad3451229aa7b9cfd1cf (diff) | |
| download | stable-diffusion-webui-gfx803-963d98639673098fa8df975dd380f1ef56fff3b5.tar.gz stable-diffusion-webui-gfx803-963d98639673098fa8df975dd380f1ef56fff3b5.tar.bz2 stable-diffusion-webui-gfx803-963d98639673098fa8df975dd380f1ef56fff3b5.zip | |
Merge branch 'AUTOMATIC1111:master' into deepdanbooru_pre_process
Diffstat (limited to 'modules/textual_inversion/learn_schedule.py')
| -rw-r--r-- | modules/textual_inversion/learn_schedule.py | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/modules/textual_inversion/learn_schedule.py b/modules/textual_inversion/learn_schedule.py new file mode 100644 index 00000000..db720271 --- /dev/null +++ b/modules/textual_inversion/learn_schedule.py @@ -0,0 +1,34 @@ +
+class LearnSchedule:
+ def __init__(self, learn_rate, max_steps, cur_step=0):
+ pairs = learn_rate.split(',')
+ self.rates = []
+ self.it = 0
+ self.maxit = 0
+ for i, pair in enumerate(pairs):
+ tmp = pair.split(':')
+ if len(tmp) == 2:
+ step = int(tmp[1])
+ if step > cur_step:
+ self.rates.append((float(tmp[0]), min(step, max_steps)))
+ self.maxit += 1
+ if step > max_steps:
+ return
+ elif step == -1:
+ self.rates.append((float(tmp[0]), max_steps))
+ self.maxit += 1
+ return
+ else:
+ self.rates.append((float(tmp[0]), max_steps))
+ self.maxit += 1
+ return
+
+ def __iter__(self):
+ return self
+
+ def __next__(self):
+ if self.it < self.maxit:
+ self.it += 1
+ return self.rates[self.it - 1]
+ else:
+ raise StopIteration
|