LICENSE
README.md
pyproject.toml
Model/Attentions.py
Model/Attentions3D.py
Model/Decoder.py
Model/Decoder3D.py
Model/Encoder.py
Model/Encoder3D.py
Model/Positional_encoder.py
Model/Positional_encoder_3d.py
Model/__init__.py
Model/eq_swin_layers.py
Model/eq_vit.py
Model/Eqpdetransformer/__init__.py
Model/Eqpdetransformer/eq_pde_blocks.py
Model/Eqpdetransformer/eq_pde_layers.py
Model/Eqpdetransformer/eq_pde_transformer.py
Model/Eqpdetransformer/none_patch_layers.py
Model/Eqpdetransformer3D/__init__.py
Model/Eqpdetransformer3D/eq_pde_blocks_3d.py
Model/Eqpdetransformer3D/eq_pde_layers_3d.py
Model/Eqpdetransformer3D/eq_pde_transformer_3d.py
Model/Eqpdetransformer3D/none_patch_layers_3d.py
config/__init__.py
config/config_ViT.py
config/config_eqpdetrans.py
config/config_eqpdetrans3d.py
config/config_helper.py
data/Adv.py
data/Apebench.py
data/P3D.py
data/ROT_dataset.py
data/Well.py
data/__init__.py
data/mnist_rot.py
data/pcam.py
lib/__init__.py
lib/data/Data_helper.py
lib/data/Local_basis.py
lib/data/Local_basis_3d_ForPICT.py
lib/data/__init__.py
lib/util/__init__.py
lib/util/argument.py
lib/util/logger.py
lib/util/logging.py
lib/util/plotting.py
lib/util/profiling.py
revit/__init__.py
revit/bases.py
revit/configs.py
revit/datasets.py
revit/models.py
revit_pde.egg-info/PKG-INFO
revit_pde.egg-info/SOURCES.txt
revit_pde.egg-info/dependency_links.txt
revit_pde.egg-info/requires.txt
revit_pde.egg-info/top_level.txt
tests/test_config_loading.py
tests/test_dataset_integration.py
tests/test_decoder_alignment.py
tests/test_model_forward.py
tests/test_public_api.py
trainer/config_search.py
trainer/model_discovery.py
trainer/rollout_utils.py
trainer/train_helper.py