↓ 6 callersFunctionbuild_mlp(
in_size,
hidden_size,
out_size,
drop_out=True,
lay_norm=True,
dropout_prob=0.2,
FVGN-src/main/FVGN_model/EncoderProcesserDecoder.py:12
↓ 3 callersMethodload_checkpoint(
self,
optimizer=None,
scheduler=None,
ckpdir=None,
device=None,
FVGN-src/main/FVGN_model/FVGN.py:328
↓ 2 callersMethod__init__(
self,
params=None,
is_traning=True,
device=None,
transform=None,
FVGN-src/main/dataset/Load_mesh.py:354
↓ 2 callersMethodcreate_next_graph(
graph_old,
graph_edge,
mask_face_boundary,
cell_next_UV,
cell_type,
FVGN-src/main/dataset/Load_mesh.py:1077
↓ 2 callersFunctionextract_mesh_state(
dataset,
tf_writer,
index,
origin_writer=None,
solving_params=None,
h5_writer=None,
FVGN-src/main/Extract_mesh/parse_tfrecord_refactor.py:419
↓ 2 callersFunctionseprate_cells(mesh_pos,cells,node_type,density,pressure,velocity,index)
MGN-src/main/meshgraphnets/parse_tfrecord_refactor.py:889
↓ 2 callersMethodupdate_cell_attr(
self,
frames,
one_hot: int,
types: torch.Tensor,
accumulation=True,
FVGN-src/main/FVGN_model/FVGN.py:83
↓ 2 callersMethodvalid_datapreprocessing(
self, graph_node, graph_edge, graph_cell, cell_noise, dual_edge=False
)
FVGN-src/main/dataset/Load_mesh.py:904
↓ 1 callersFunctionparse_node_center_to_cell_center(traj_data,rollout_index,uv_MSE,plot_boundary_p_time_step=None,save_tec=False,plot_boundary=False,traj_idx=Non
MGN-src/main/run_model.py:290
↓ 1 callersFunctionrecover_ghosted_2_fore_mesh(
ghosted_domain,
mode="cylinder_mesh",
unorder=False,
limit=None,
index=None,
fig=Non
FVGN-src/main/Extract_mesh/parse_tfrecord_refactor.py:2455
↓ 1 callersFunctionrollout(
model,
datasets,
batch_index=[],
result_dir=None,
rollout_start_step=None,
rollout_t
FVGN-src/main/rollout.py:229