Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
S
Stable Diffusion Webui
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Locked Files
Issues
0
Issues
0
List
Boards
Labels
Service Desk
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Security & Compliance
Security & Compliance
Dependency List
License Compliance
Packages
Packages
List
Container Registry
Analytics
Analytics
CI / CD
Code Review
Insights
Issues
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
novelai-storage
Stable Diffusion Webui
Commits
8636b50a
Commit
8636b50a
authored
Oct 13, 2022
by
Melan
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Add learn_rate to csv and removed a left-over debug statement
parent
1cfc2a18
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
6 additions
and
5 deletions
+6
-5
modules/hypernetworks/hypernetwork.py
modules/hypernetworks/hypernetwork.py
+3
-3
modules/textual_inversion/textual_inversion.py
modules/textual_inversion/textual_inversion.py
+3
-2
No files found.
modules/hypernetworks/hypernetwork.py
View file @
8636b50a
...
@@ -257,19 +257,19 @@ def train_hypernetwork(hypernetwork_name, learn_rate, data_root, log_directory,
...
@@ -257,19 +257,19 @@ def train_hypernetwork(hypernetwork_name, learn_rate, data_root, log_directory,
last_saved_file
=
os
.
path
.
join
(
hypernetwork_dir
,
f
'{hypernetwork_name}-{hypernetwork.step}.pt'
)
last_saved_file
=
os
.
path
.
join
(
hypernetwork_dir
,
f
'{hypernetwork_name}-{hypernetwork.step}.pt'
)
hypernetwork
.
save
(
last_saved_file
)
hypernetwork
.
save
(
last_saved_file
)
print
(
f
"{write_csv_every} > {hypernetwork.step
%
write_csv_every == 0}, {write_csv_every}"
)
if
write_csv_every
>
0
and
hypernetwork_dir
is
not
None
and
hypernetwork
.
step
%
write_csv_every
==
0
:
if
write_csv_every
>
0
and
hypernetwork_dir
is
not
None
and
hypernetwork
.
step
%
write_csv_every
==
0
:
write_csv_header
=
False
if
os
.
path
.
exists
(
os
.
path
.
join
(
hypernetwork_dir
,
"hypernetwork_loss.csv"
))
else
True
write_csv_header
=
False
if
os
.
path
.
exists
(
os
.
path
.
join
(
hypernetwork_dir
,
"hypernetwork_loss.csv"
))
else
True
with
open
(
os
.
path
.
join
(
hypernetwork_dir
,
"hypernetwork_loss.csv"
),
"a+"
)
as
fout
:
with
open
(
os
.
path
.
join
(
hypernetwork_dir
,
"hypernetwork_loss.csv"
),
"a+"
)
as
fout
:
csv_writer
=
csv
.
DictWriter
(
fout
,
fieldnames
=
[
"step"
,
"loss"
])
csv_writer
=
csv
.
DictWriter
(
fout
,
fieldnames
=
[
"step"
,
"loss"
,
"learn_rate"
])
if
write_csv_header
:
if
write_csv_header
:
csv_writer
.
writeheader
()
csv_writer
.
writeheader
()
csv_writer
.
writerow
({
"step"
:
hypernetwork
.
step
,
csv_writer
.
writerow
({
"step"
:
hypernetwork
.
step
,
"loss"
:
f
"{losses.mean():.7f}"
})
"loss"
:
f
"{losses.mean():.7f}"
,
"learn_rate"
:
scheduler
.
learn_rate
})
if
hypernetwork
.
step
>
0
and
images_dir
is
not
None
and
hypernetwork
.
step
%
create_image_every
==
0
:
if
hypernetwork
.
step
>
0
and
images_dir
is
not
None
and
hypernetwork
.
step
%
create_image_every
==
0
:
last_saved_image
=
os
.
path
.
join
(
images_dir
,
f
'{hypernetwork_name}-{hypernetwork.step}.png'
)
last_saved_image
=
os
.
path
.
join
(
images_dir
,
f
'{hypernetwork_name}-{hypernetwork.step}.png'
)
...
...
modules/textual_inversion/textual_inversion.py
View file @
8636b50a
...
@@ -262,14 +262,15 @@ def train_embedding(embedding_name, learn_rate, data_root, log_directory, traini
...
@@ -262,14 +262,15 @@ def train_embedding(embedding_name, learn_rate, data_root, log_directory, traini
with
open
(
os
.
path
.
join
(
log_directory
,
"textual_inversion_loss.csv"
),
"a+"
)
as
fout
:
with
open
(
os
.
path
.
join
(
log_directory
,
"textual_inversion_loss.csv"
),
"a+"
)
as
fout
:
csv_writer
=
csv
.
DictWriter
(
fout
,
fieldnames
=
[
"epoch"
,
"epoch_step"
,
"loss"
])
csv_writer
=
csv
.
DictWriter
(
fout
,
fieldnames
=
[
"epoch"
,
"epoch_step"
,
"loss"
,
"learn_rate"
])
if
write_csv_header
:
if
write_csv_header
:
csv_writer
.
writeheader
()
csv_writer
.
writeheader
()
csv_writer
.
writerow
({
"epoch"
:
epoch_num
+
1
,
csv_writer
.
writerow
({
"epoch"
:
epoch_num
+
1
,
"epoch_step"
:
epoch_step
-
1
,
"epoch_step"
:
epoch_step
-
1
,
"loss"
:
f
"{losses.mean():.7f}"
})
"loss"
:
f
"{losses.mean():.7f}"
,
"learn_rate"
:
scheduler
.
learn_rate
})
if
embedding
.
step
>
0
and
images_dir
is
not
None
and
embedding
.
step
%
create_image_every
==
0
:
if
embedding
.
step
>
0
and
images_dir
is
not
None
and
embedding
.
step
%
create_image_every
==
0
:
last_saved_image
=
os
.
path
.
join
(
images_dir
,
f
'{embedding_name}-{embedding.step}.png'
)
last_saved_image
=
os
.
path
.
join
(
images_dir
,
f
'{embedding_name}-{embedding.step}.png'
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment