Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dgl
Commits
ff66c42b
Unverified
Commit
ff66c42b
authored
Feb 28, 2024
by
Andrei Ivanov
Committed by
GitHub
Feb 29, 2024
Browse files
[GCMC]Fixing warnings that appear during in `gcmc` training. (#7152)
parent
85c030f6
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
7 deletions
+7
-7
examples/pytorch/gcmc/train.py
examples/pytorch/gcmc/train.py
+7
-7
No files found.
examples/pytorch/gcmc/train.py
View file @
ff66c42b
...
@@ -194,14 +194,14 @@ def train(args):
...
@@ -194,14 +194,14 @@ def train(args):
loss
=
count_loss
/
(
iter_idx
+
1
),
loss
=
count_loss
/
(
iter_idx
+
1
),
rmse
=
count_rmse
/
count_num
,
rmse
=
count_rmse
/
count_num
,
)
)
logging_str
=
(
logging_str
=
"Iter={:4d}, loss={:.4f}, rmse={:.4f}"
.
format
(
"Iter={}, loss={:.4f}, rmse={:.4f}, time={:.4f}"
.
format
(
iter_idx
,
iter_idx
,
count_loss
/
iter_idx
,
count_loss
/
iter_idx
,
count_rmse
/
count_num
,
count_rmse
/
count_num
,
np
.
average
(
dur
),
)
)
)
if
iter_idx
>
3
:
logging_str
+=
", time={:.4f}"
.
format
(
np
.
average
(
dur
))
count_rmse
=
0
count_rmse
=
0
count_num
=
0
count_num
=
0
...
...
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