Open NohTow opened 4 months ago
Right now, the normalization is done "by hand" in the encode function or need to be done afterwards when the forward pass is used. It would be better to use the "Normalize" module from ST (this would require a bit of change in the logic of loading).
Right now, the normalization is done "by hand" in the encode function or need to be done afterwards when the forward pass is used. It would be better to use the "Normalize" module from ST (this would require a bit of change in the logic of loading).