mdribeiro / DeepCFD

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

Layers per block in DeepCFD.py #5

Open pomtojoer opened 3 years ago

pomtojoer commented 3 years ago

Hi there,

Thank you for publishing this! I have been looking through the code and noticed that the number of layers per block was not specified in your DeepCFD.py script.

model = UNetEx(3, 3, filters=filters, kernel_size=kernel_size, batch_norm=bn, weight_norm=wn)

This results in the layers per block defaulting to 3 as defined in UNetEx class. I was just wondering if this was intended as this differs from your paper's implementation of 2 layers per block. Thank you again for this!

Kind regards, Sean

mdribeiro commented 3 years ago

Hi Sean,

Thank you very much for your interest in our work.

That was intended so. The default of 3 was chosen after the last run of hyperparameter/architecture search, which was executed after our initial runs during the writing of the paper.

I hope this clarifies this point.

Best, Mateus