2020-06-27 14:26:57 +00:00
|
|
|
from torch.utils.tensorboard import SummaryWriter
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
2020-07-01 17:28:23 +00:00
|
|
|
writer = SummaryWriter("../experiments/train_div2k_feat_nsgen_r3/recovered_tb")
|
|
|
|
|
f = open("../experiments/train_div2k_feat_nsgen_r3/console.txt", encoding="utf8")
|
2020-06-27 14:26:57 +00:00
|
|
|
console = f.readlines()
|
|
|
|
|
search_terms = [
|
|
|
|
|
("iter", ", iter: ", ", lr:"),
|
|
|
|
|
("l_g_total", " l_g_total: ", " switch_temperature:")
|
|
|
|
|
]
|
|
|
|
|
iter = 0
|
|
|
|
|
for line in console:
|
|
|
|
|
if " - INFO: [epoch:" not in line:
|
|
|
|
|
continue
|
|
|
|
|
for name, start, end in search_terms:
|
|
|
|
|
val = line[line.find(start)+len(start):line.find(end)].replace(",", "")
|
|
|
|
|
if name == "iter":
|
|
|
|
|
iter = int(val)
|
|
|
|
|
else:
|
|
|
|
|
writer.add_scalar(name, float(val), iter)
|
|
|
|
|
writer.close()
|
