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): ...@@ -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
......
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