search

mdribeiro / DeepCFD

DeepCFD: Efficient Steady-State Laminar Flow Approximation with Deep Convolutional Neural Networks
MIT License
187 stars 62 forks source link

Final Trained Weights #3

Open joshuamills98 opened 3 years ago

joshuamills98 commented 3 years ago

Hi there, Thanks for uploading this, I am exploring the code now and was wondering if you could provide the finalized weights of the model. I don't have the CPU to run the training unfortunately but would like to explore the result. Josh

Monica-Xiao commented 3 years ago

after you edit the file path of torch.save() in pytorchtools.py. and import this file in the DeepCFD.py. you can call the torch.save() to save the checkpoint file after you run your training