search

AndreWeiner / ml-cfd-lecture

Lecture material for machine learning applied to computational fluid mechanics
GNU General Public License v3.0
328 stars 115 forks source link

Revision of lecture/exercise bubble path classification #27

Closed AndreWeiner closed 1 year ago

AndreWeiner commented 1 year ago

Hi @JanisGeise, I just updated the classification exercise (exercise 6). Please let me know if changes/additions are required. I uploaded a reference solution to your cloud folder. Thanks! Andre

JanisGeise commented 1 year ago

Hi @AndreWeiner,

the exercise worked without any problems. Recently, I get warnings from PyTorch regarding Cuda, maybe (in case other people might get this warning too) you can add a note to ignore such warnings, since the training is done on CPU.

Out of curiosity, in the last cell of the lecture notebook (cell 90) you add 0.01 to the predictions (prediction = pt.argmax(prob, dim=1) + 0.01), is this just for plotting purposes?

Regards Janis

AndreWeiner commented 1 year ago

Perfect, thanks for the feedback. I got the warning about Cuda, too. Not sure what it is about. I thought it was only related to my system. Yes, the little increment is just for plotting - I think to avoid the influence of roundoff errors in the Matplotlib ColorMap. In the current version of the notebook, it also works for me without adding the increment. Thanks again! Andre

JanisGeise commented 1 year ago

Ok, it's good to know that this seems not to be just an issue with my torch installation / setup. Thanks for the explanation regarding the increment.

Regards, Janis