forked from mrq/DL-Art-School
21 lines
779 B
Python
21 lines
779 B
Python
|
from torch.utils.tensorboard import SummaryWriter
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
writer = SummaryWriter("../experiments/train_div2k_feat_rg2_more_stuff")
|
||
|
|
f = open("../experiments/train_div2k_feat_rg2_more_stuff/console_output")
|
||
|
|
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()
|