1a5d24a4dcd75c0056fcc5a0ca6f529ddaaa0160,models/varnet/varnet.py,VariationalNetworkModel,test_step,#VariationalNetworkModel#,256

Before Change

    def test_step(self, batch, batch_idx):
        input, _, mean, std, fname, slice = batch
        output = self.forward(input)
        mean = mean.unsqueeze(1).unsqueeze(2)
        std = std.unsqueeze(1).unsqueeze(2)
        return {
            "fname": fname,
            "slice": slice,

After Change

    def test_step(self, batch, batch_idx):
        masked_kspace, mask, target, fname, slice, _ = batch
        output = self.forward(masked_kspace, mask)
        _, output = T.center_crop_to_smallest(target, output)
        return {
            "fname": fname,
            "slice": slice,

Instances