File size: 445 Bytes
bdce880 | 1 2 3 4 5 6 7 8 9 10 11 | from model import Transolver_Irregular_Mesh, Transolver_Structured_Mesh_2D, Transolver_Structured_Mesh_3D
def get_model(args):
model_dict = {
'Transolver_Irregular_Mesh': Transolver_Irregular_Mesh, # for PDEs in 1D space or in unstructured meshes
'Transolver_Structured_Mesh_2D': Transolver_Structured_Mesh_2D,
'Transolver_Structured_Mesh_3D': Transolver_Structured_Mesh_3D,
}
return model_dict[args.model]
|