Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
598f7fcd
S
Stable Diffusion Webui
项目概览
Overbill1683
/
Stable Diffusion Webui
大约 1 年 前同步成功
通知
1786
Star
81
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
分析
仓库
DevOps
项目成员
Pages
S
Stable Diffusion Webui
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Pages
分析
分析
仓库分析
DevOps
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
提交
体验新版 GitCode,发现更多精彩内容 >>
提交
598f7fcd
编写于
1月 16, 2023
作者:
A
aria1th
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix loss_dict problem
上级
205991df
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
3 addition
and
1 deletion
+3
-1
modules/hypernetworks/hypernetwork.py
modules/hypernetworks/hypernetwork.py
+3
-1
未找到文件。
modules/hypernetworks/hypernetwork.py
浏览文件 @
598f7fcd
...
...
@@ -561,6 +561,7 @@ def train_hypernetwork(id_task, hypernetwork_name, learn_rate, batch_size, gradi
_loss_step
=
0
#internal
# size = len(ds.indexes)
# loss_dict = defaultdict(lambda : deque(maxlen = 1024))
loss_logging
=
[]
# losses = torch.zeros((size,))
# previous_mean_losses = [0]
# previous_mean_loss = 0
...
...
@@ -601,6 +602,7 @@ def train_hypernetwork(id_task, hypernetwork_name, learn_rate, batch_size, gradi
else
:
c
=
stack_conds
(
batch
.
cond
).
to
(
devices
.
device
,
non_blocking
=
pin_memory
)
loss
=
shared
.
sd_model
(
x
,
c
)[
0
]
/
gradient_step
loss_logging
.
append
(
loss
.
item
())
del
x
del
c
...
...
@@ -644,7 +646,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
()
)
mean_loss
=
sum
(
loss_logging
)
/
len
(
loss_logging
)
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
,
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录