Unverified Commit ff66c42b authored by Andrei Ivanov's avatar Andrei Ivanov Committed by GitHub
Browse files

[GCMC]Fixing warnings that appear during in `gcmc` training. (#7152)

parent 85c030f6
......@@ -194,14 +194,14 @@ def train(args):
loss=count_loss / (iter_idx + 1),
rmse=count_rmse / count_num,
)
logging_str = (
"Iter={}, loss={:.4f}, rmse={:.4f}, time={:.4f}".format(
iter_idx,
count_loss / iter_idx,
count_rmse / count_num,
np.average(dur),
)
logging_str = "Iter={:4d}, loss={:.4f}, rmse={:.4f}".format(
iter_idx,
count_loss / iter_idx,
count_rmse / count_num,
)
if iter_idx > 3:
logging_str += ", time={:.4f}".format(np.average(dur))
count_rmse = 0
count_num = 0
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment