Hello iamalexkorotin :-)
I'm an undergraduate student majored in computer science.
Before mentioning what I need, firstly your code was very helpful for me.
I'm working on a problem which is how to solve optimal transport efficiently.
I find that in your notebooks, both train_hd and test_hd containing the training process.
I wonder how can I test the model after training, which is to make the model generate a transport map directly.
And could you show me the test code?
Thanks very much!