shashanksubramanian/neuraloperators-tl-scaling

Code for Characterizing Scaling and Transfer Learning Behavior of FNO in SciML

neuraloperators-tl-scaling's Issues

Scales for solution function not used?

The median value for the solution functions is computed in get_scales.ipynb, but is never used in the following normalization code:

if self.scales is not None:
            f_norm = np.linalg.norm(X[0]) * self.measure
            f_scaling = f_norm / self.scales[0]
            X = X / f_scaling # ensures that 10f and 10k for example, have the same input
            # scale the tensors
            X[self.in_channels:] = X[self.in_channels:] / self.scales[self.in_channels:(self.in_channels + self.tensor_shape), None, None]

Is this intentional, or accidental?