Forward function used in test time. This function wraps function in no_grad() so we don't save intermediate steps for backprop It also calls to produce additional visualization results
(self)
| 96 | net.eval() |
| 97 | |
| 98 | def test(self): |
| 99 | """Forward function used in test time. |
| 100 | |
| 101 | This function wraps <forward> function in no_grad() so we don't save intermediate steps for backprop |
| 102 | It also calls <compute_visuals> to produce additional visualization results |
| 103 | """ |
| 104 | with torch.no_grad(): |
| 105 | self.forward() |
| 106 | self.compute_visuals() |
| 107 | |
| 108 | def compute_visuals(self): |
| 109 | """Calculate additional output images for visdom and HTML visualization""" |
no test coverage detected