(self, scale=1.0)
| 757 | # print(self.FetchBlobWrapper(self.tint)) |
| 758 | |
| 759 | def MSEloss(self, scale=1.0): |
| 760 | # add MSEloss to the model |
| 761 | self.AddLayerWrapper(self.model.SquaredL2Distance, [self.tout, self.ttar], "sd") |
| 762 | self.AddLayerWrapper(self.model.Scale, "sd", "sd2", scale=2.0 * scale) |
| 763 | # WARNING: "loss" is a special tag and should not be changed |
| 764 | self.loss = self.AddLayerWrapper(self.model.AveragedLoss, "sd2", "loss") |
| 765 | |
| 766 | def BCEloss(self, scale=1.0, threshold=0.0): |
| 767 | # add BCEloss to the mode |
no test coverage detected