Alison

Another improvement would be to check if the size are as expected. For example, in `DiscretizedManifold.fit()` function: ``` with torch.no_grad(): weight = model.curve_length(line(t)) assert weight.shape == bs, f"model.curve_length should return...

Another question regarding `DiscretizedManifold.fit()`: a graph is created based on two points obtained from a curve: ``` t = torch.linspace(0, 1, 2) (...) with torch.no_grad(): weight = model.curve_length(line(t)) ``` and...