self.img('input_img',t.Tensor(64,64))
(self, name, img_)
| 971 | opts=self.opts) |
| 972 | |
| 973 | def img(self, name, img_): |
| 974 | """ |
| 975 | self.img('input_img',t.Tensor(64,64)) |
| 976 | """ |
| 977 | |
| 978 | if len(img_.size()) < 3: |
| 979 | img_ = img_.cpu().unsqueeze(0) |
| 980 | self._viz.image(img_.cpu(), |
| 981 | win=name, |
| 982 | opts=dict(title=name) |
| 983 | ) |
| 984 | def img_grid_many(self, d): |
| 985 | for k, v in d.items(): |
| 986 | self.img_grid(k, v) |