Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in / Register
Toggle navigation
S
stable-diffusion-webui
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Administrator
stable-diffusion-webui
Commits
205991df
Commit
205991df
authored
Jan 15, 2023
by
AUTOMATIC
Browse files
Options
Browse Files
Download
Plain Diff
Merge remote-tracking branch 'origin/fix-mean-loss'
parents
b6ce041c
16f41089
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
hypernetwork.py
modules/hypernetworks/hypernetwork.py
+1
-1
No files found.
modules/hypernetworks/hypernetwork.py
View file @
205991df
...
...
@@ -644,7 +644,7 @@ def train_hypernetwork(id_task, hypernetwork_name, learn_rate, batch_size, gradi
if
shared
.
opts
.
training_enable_tensorboard
:
epoch_num
=
hypernetwork
.
step
//
len
(
ds
)
epoch_step
=
hypernetwork
.
step
-
(
epoch_num
*
len
(
ds
))
+
1
mean_loss
=
sum
(
sum
(
x
)
for
x
in
loss_dict
.
values
())
/
sum
(
len
(
x
)
for
x
in
loss_dict
.
values
())
textual_inversion
.
tensorboard_add
(
tensorboard_writer
,
loss
=
mean_loss
,
global_step
=
hypernetwork
.
step
,
step
=
epoch_step
,
learn_rate
=
scheduler
.
learn_rate
,
epoch_num
=
epoch_num
)
textual_inversion
.
write_loss
(
log_directory
,
"hypernetwork_loss.csv"
,
hypernetwork
.
step
,
steps_per_epoch
,
{
...
...
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