Visualising loss in weight space

There has been plenty of talk in the lectures about how using the gradient descent we move through the weight space (or feature space) to minimise the loss. Usually, during training and optimisation, I’m using only parameters like accuracy, F1 to better my model.

Is there a way to visualise loss in weight space using fastai or any other library?