↓ 1 callersFunctionseprate_cells(mesh_pos, cells, node_type, density, pressure, velocity, index)
FVGN-src/main/Extract_mesh/parse_tfrecord_refactor.py:1153
Method__init__(
self, optimizer, decay_steps=10000, gamma=0.4, min_lr=1e-7, last_epoch=-1
)
FVGN-src/main/utils/scheduler.py:149
Method__init__(
self,
optimizer,
start_lr,
drop_lr,
total_epochs,
gamma,
FVGN-src/main/utils/scheduler.py:179
Method__init__(
self,
mesh_file,
data_file,
tf_writer=None,
h5_writer=None,
FVGN-src/main/Extract_mesh/parse_comsol.py:22
Method__init__(
self,
params,
path,
split,
description=description,
max_epoc
FVGN-src/main/dataset/Load_mesh.py:207
Method__init__(
self,
hidden_size=128,
drop_out=False,
mp_times=2,
MultiHead=1,
FVGN-src/main/FVGN_model/EncoderProcesserDecoder.py:81
Method__init__(
self,
input_size,
attention_size,
mp_times=2,
MultiHead=1,
c
FVGN-src/main/FVGN_model/GN_blocks.py:59
Method__init__(
self,
output_size,
latent_size,
num_layers,
message_passing_steps,
MGN-src/main/meshgraphnets/core_model.py:97
Functiondirect_decode_loss(
params,
graph_cell,
graph_edge,
mask_face_interior,
predicted_edge_attr,
target_face
FVGN-src/main/utils/loss_compute.py:5
Methodforward(
self,
uv_face=0,
p_face=0,
flux_D=0,
unv=None,
rho=None,
FVGN-src/main/FVGN_model/EncoderProcesserDecoder.py:251