Write visaul examples to the event file
(prn_net: PlaneRecNet, val_dataset: PlaneAnnoDataset, writer: SummaryWriter, iteration, eval_nums=5)
| 461 | |
| 462 | |
| 463 | def log_visual_example(prn_net: PlaneRecNet, val_dataset: PlaneAnnoDataset, writer: SummaryWriter, iteration, eval_nums=5): |
| 464 | """ |
| 465 | Write visaul examples to the event file |
| 466 | """ |
| 467 | with torch.no_grad(): |
| 468 | prn_net.eval() |
| 469 | start = time.time() |
| 470 | eval_script.tensorborad_visual_log(prn_net, val_dataset, writer, iteration, eval_nums) |
| 471 | end = time.time() |
| 472 | prn_net.train() |
| 473 | |
| 474 | |
| 475 | if __name__ == "__main__": |