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)
| 117 | net.eval() |
| 118 | |
| 119 | def test(self): |
| 120 | """Forward function used in test time. |
| 121 | |
| 122 | This function wraps <forward> function in no_grad() so we don't save intermediate steps for backprop |
| 123 | It also calls <compute_visuals> to produce additional visualization results |
| 124 | """ |
| 125 | with torch.no_grad(): |
| 126 | self.forward() |
| 127 | self.compute_visuals() |
| 128 | |
| 129 | def compute_visuals(self): |
| 130 | """Calculate additional output images for visdom and HTML visualization""" |
no test coverage detected