Transformer-PhysX
latest
Getting Started:
Installation
Quick Start
Python API:
trphysx.config
trphysx.data_utils
trphysx.embedding
trphysx.transformer
trphysx.utils
trphysx.viz
Additional Information:
Contact
License
Transformer-PhysX
»
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
A
appendToHDF5() (trphysx.utils.metrics.Metrics method)
ArgUtils (class in trphysx.config.args)
AutoDataHandler (class in trphysx.embedding.training.enn_data_handler)
AutoDataset (class in trphysx.data_utils.dataset_auto)
AutoEmbeddingModel (class in trphysx.embedding.embedding_auto)
AutoPhysConfig (class in trphysx.config.configuration_auto)
AutoViz (class in trphysx.viz.viz_auto)
B
Block (class in trphysx.transformer.phys_transformer_gpt2)
block_size (trphysx.config.args.TrainingArguments attribute)
C
cache_path (trphysx.config.args.DataArguments attribute)
ckpt_dir (trphysx.config.args.TrainingArguments attribute)
config() (trphysx.config.args.ArgUtils class method)
config_name (trphysx.config.args.ModelArguments attribute)
configModelNames() (trphysx.config.args.ArgUtils class method)
configPaths() (trphysx.config.args.ArgUtils class method)
configTorchDevices() (trphysx.config.args.ArgUtils class method)
Conv1D (class in trphysx.transformer.utils)
create_artists() (trphysx.viz.viz_lorenz.HandlerColormap method)
create_dataset() (trphysx.data_utils.dataset_auto.AutoDataset class method)
createTestingLoader() (trphysx.embedding.training.enn_data_handler.CylinderDataHandler method)
(trphysx.embedding.training.enn_data_handler.EmbeddingDataHandler method)
(trphysx.embedding.training.enn_data_handler.GrayScottDataHandler method)
(trphysx.embedding.training.enn_data_handler.LorenzDataHandler method)
createTrainingLoader() (trphysx.embedding.training.enn_data_handler.CylinderDataHandler method)
(trphysx.embedding.training.enn_data_handler.EmbeddingDataHandler method)
(trphysx.embedding.training.enn_data_handler.GrayScottDataHandler method)
(trphysx.embedding.training.enn_data_handler.LorenzDataHandler method)
CylinderConfig (class in trphysx.config.configuration_cylinder)
CylinderDataHandler (class in trphysx.embedding.training.enn_data_handler)
CylinderDataHandler.CylinderDataCollator (class in trphysx.embedding.training.enn_data_handler)
CylinderDataHandler.CylinderDataset (class in trphysx.embedding.training.enn_data_handler)
CylinderDataset (class in trphysx.data_utils.dataset_cylinder)
CylinderEmbedding (class in trphysx.embedding.embedding_cylinder)
CylinderEmbeddingTrainer (class in trphysx.embedding.embedding_cylinder)
CylinderViz (class in trphysx.viz.viz_cylinder)
D
DataArguments (class in trphysx.config.args)
DataClass (class in trphysx.config.arg_parser)
DataCollator (class in trphysx.data_utils.data_utils)
dataloader_drop_last (trphysx.config.args.TrainingArguments attribute)
delHDF5() (trphysx.utils.metrics.Metrics method)
devices (trphysx.embedding.embedding_model.EmbeddingModel attribute)
E
embed() (trphysx.embedding.embedding_cylinder.CylinderEmbedding method)
(trphysx.embedding.embedding_grayscott.GrayScottEmbedding method)
(trphysx.embedding.embedding_lorenz.LorenzEmbedding method)
(trphysx.embedding.embedding_model.EmbeddingModel method)
embed_data() (trphysx.data_utils.dataset_cylinder.CylinderDataset method)
(trphysx.data_utils.dataset_grayscott.GrayscottDataset method)
(trphysx.data_utils.dataset_lorenz.LorenzDataset method)
(trphysx.data_utils.dataset_phys.PhysicalDataset method)
embed_dims (trphysx.embedding.embedding_model.EmbeddingModel attribute)
embedding_file_or_path (trphysx.config.args.ModelArguments attribute)
embedding_name (trphysx.config.args.ModelArguments attribute)
EmbeddingDataHandler (class in trphysx.embedding.training.enn_data_handler)
EmbeddingModel (class in trphysx.embedding.embedding_model)
EmbeddingParser (class in trphysx.embedding.training.enn_args)
EmbeddingTrainer (class in trphysx.embedding.training.enn_trainer)
EmbeddingTrainingHead (class in trphysx.embedding.embedding_model)
epoch_start (trphysx.config.args.TrainingArguments attribute)
epochs (trphysx.config.args.TrainingArguments attribute)
eval_batch_size (trphysx.config.args.TrainingArguments attribute)
eval_h5_file (trphysx.config.args.DataArguments attribute)
eval_states() (trphysx.utils.trainer.Trainer method)
eval_step() (trphysx.utils.trainer.Trainer method)
eval_steps (trphysx.config.args.TrainingArguments attribute)
evaluate() (trphysx.embedding.embedding_cylinder.CylinderEmbeddingTrainer method)
(trphysx.embedding.embedding_lorenz.LorenzEmbeddingTrainer method)
(trphysx.embedding.embedding_model.EmbeddingTrainingHead method)
(trphysx.embedding.training.enn_trainer.EmbeddingTrainer method)
(trphysx.transformer.phys_transformer_helpers.PhysformerTrain method)
(trphysx.utils.trainer.Trainer method)
exp_dir (trphysx.config.args.TrainingArguments attribute)
F
file_name (trphysx.utils.metrics.Metrics attribute)
file_path (trphysx.utils.metrics.Metrics attribute)
forward() (trphysx.embedding.embedding_cylinder.CylinderEmbedding method)
(trphysx.embedding.embedding_cylinder.CylinderEmbeddingTrainer method)
(trphysx.embedding.embedding_grayscott.GrayScottEmbedding method)
(trphysx.embedding.embedding_grayscott.GrayScottEmbeddingTrainer method)
(trphysx.embedding.embedding_lorenz.LorenzEmbedding method)
(trphysx.embedding.embedding_lorenz.LorenzEmbeddingTrainer method)
(trphysx.embedding.embedding_model.EmbeddingTrainingHead method)
(trphysx.transformer.attention.MaskedAttention method)
(trphysx.transformer.phys_transformer_base.PhysformerBase method)
(trphysx.transformer.phys_transformer_gpt2.Block method)
(trphysx.transformer.phys_transformer_gpt2.MLP method)
(trphysx.transformer.phys_transformer_gpt2.PhysformerGPT2 method)
(trphysx.transformer.phys_transformer_helpers.PhysformerTrain method)
(trphysx.transformer.utils.Conv1D method)
from_dict() (trphysx.config.configuration_phys.PhysConfig class method)
from_json_file() (trphysx.config.configuration_auto.AutoPhysConfig class method)
G
gelu_fast() (in module trphysx.transformer.utils)
gelu_new() (in module trphysx.transformer.utils)
generate() (trphysx.transformer.generate_utils.GenerationMixin method)
(trphysx.transformer.phys_transformer_base.PhysformerBase method)
(trphysx.transformer.phys_transformer_helpers.PhysformerTrain method)
GenerationMixin (class in trphysx.transformer.generate_utils)
get_activation() (in module trphysx.transformer.utils)
get_eval_dataloader() (trphysx.utils.trainer.Trainer method)
get_input_embeddings() (trphysx.transformer.phys_transformer_base.PhysformerBase method)
get_train_dataloader() (trphysx.utils.trainer.Trainer method)
gradient_accumulation_steps (trphysx.config.args.TrainingArguments attribute)
GrayScottConfig (class in trphysx.config.configuration_grayscott)
GrayScottDataHandler (class in trphysx.embedding.training.enn_data_handler)
GrayScottDataHandler.GrayScottDataCollator (class in trphysx.embedding.training.enn_data_handler)
GrayScottDataHandler.GrayScottDataset (class in trphysx.embedding.training.enn_data_handler)
GrayscottDataset (class in trphysx.data_utils.dataset_grayscott)
GrayScottEmbedding (class in trphysx.embedding.embedding_grayscott)
GrayScottEmbeddingTrainer (class in trphysx.embedding.embedding_grayscott)
GrayscottPredictDataset (class in trphysx.data_utils.dataset_grayscott)
GrayScottViz (class in trphysx.viz.viz_grayscott)
H
HandlerColormap (class in trphysx.viz.viz_lorenz)
HfArgumentParser (class in trphysx.config.arg_parser)
hidden_size (trphysx.config.configuration_cylinder.CylinderConfig attribute)
(trphysx.config.configuration_grayscott.GrayScottConfig attribute)
(trphysx.config.configuration_lorenz.LorenzConfig attribute)
I
init_model() (trphysx.embedding.embedding_auto.AutoEmbeddingModel class method)
init_name (trphysx.config.args.ModelArguments attribute)
init_trainer() (trphysx.embedding.embedding_auto.AutoEmbeddingModel class method)
input_dims (trphysx.embedding.embedding_model.EmbeddingModel attribute)
K
koopmanDiag (trphysx.embedding.embedding_cylinder.CylinderEmbedding attribute)
(trphysx.embedding.embedding_grayscott.GrayScottEmbedding attribute)
(trphysx.embedding.embedding_lorenz.LorenzEmbedding attribute)
(trphysx.embedding.embedding_model.EmbeddingModel attribute)
koopmanOperation() (trphysx.embedding.embedding_cylinder.CylinderEmbedding method)
(trphysx.embedding.embedding_grayscott.GrayScottEmbedding method)
(trphysx.embedding.embedding_lorenz.LorenzEmbedding method)
koopmanOperator (trphysx.embedding.embedding_cylinder.CylinderEmbedding attribute)
(trphysx.embedding.embedding_grayscott.GrayScottEmbedding attribute)
(trphysx.embedding.embedding_lorenz.LorenzEmbedding attribute)
(trphysx.embedding.embedding_model.EmbeddingModel attribute)
L
linear_act() (in module trphysx.transformer.utils)
load_config() (trphysx.config.configuration_auto.AutoPhysConfig class method)
load_data_handler() (trphysx.embedding.training.enn_data_handler.AutoDataHandler class method)
load_model() (trphysx.embedding.embedding_auto.AutoEmbeddingModel class method)
(trphysx.embedding.embedding_model.EmbeddingModel method)
(trphysx.embedding.embedding_model.EmbeddingTrainingHead method)
(trphysx.transformer.phys_transformer_base.PhysformerBase method)
(trphysx.transformer.phys_transformer_helpers.PhysformerTrain method)
load_viz() (trphysx.viz.viz_auto.AutoViz class method)
local_rank (trphysx.config.args.TrainingArguments attribute)
LorenzConfig (class in trphysx.config.configuration_lorenz)
LorenzDataHandler (class in trphysx.embedding.training.enn_data_handler)
LorenzDataHandler.LorenzDataCollator (class in trphysx.embedding.training.enn_data_handler)
LorenzDataHandler.LorenzDataset (class in trphysx.embedding.training.enn_data_handler)
LorenzDataset (class in trphysx.data_utils.dataset_lorenz)
LorenzEmbedding (class in trphysx.embedding.embedding_lorenz)
LorenzEmbeddingTrainer (class in trphysx.embedding.embedding_lorenz)
LorenzViz (class in trphysx.viz.viz_lorenz)
lr (trphysx.config.args.TrainingArguments attribute)
M
MaskedAttention (class in trphysx.transformer.attention)
max_grad_norm (trphysx.config.args.TrainingArguments attribute)
merge_heads() (trphysx.transformer.attention.MaskedAttention method)
Metrics (class in trphysx.utils.metrics)
mish() (in module trphysx.transformer.utils)
mkdirs() (trphysx.embedding.training.enn_args.EmbeddingParser method)
MLP (class in trphysx.transformer.phys_transformer_gpt2)
model_name (trphysx.config.args.ModelArguments attribute)
(trphysx.embedding.embedding_cylinder.CylinderEmbedding attribute)
(trphysx.embedding.embedding_grayscott.GrayScottEmbedding attribute)
(trphysx.embedding.embedding_lorenz.LorenzEmbedding attribute)
(trphysx.embedding.embedding_model.EmbeddingModel attribute)
(trphysx.transformer.phys_transformer_base.PhysformerBase attribute)
model_type (trphysx.config.configuration_cylinder.CylinderConfig attribute)
(trphysx.config.configuration_grayscott.GrayScottConfig attribute)
(trphysx.config.configuration_lorenz.LorenzConfig attribute)
(trphysx.config.configuration_phys.PhysConfig attribute)
ModelArguments (class in trphysx.config.args)
mu (trphysx.embedding.training.enn_data_handler.EmbeddingDataHandler attribute)
N
n_eval (trphysx.config.args.DataArguments attribute)
n_gpu (trphysx.config.args.TrainingArguments attribute)
n_train (trphysx.config.args.DataArguments attribute)
norm_params (trphysx.embedding.training.enn_data_handler.EmbeddingDataHandler attribute)
notes (trphysx.config.args.TrainingArguments attribute)
num_attention_heads (trphysx.config.configuration_cylinder.CylinderConfig attribute)
(trphysx.config.configuration_grayscott.GrayScottConfig attribute)
(trphysx.config.configuration_lorenz.LorenzConfig attribute)
num_hidden_layers (trphysx.config.configuration_cylinder.CylinderConfig attribute)
(trphysx.config.configuration_grayscott.GrayScottConfig attribute)
(trphysx.config.configuration_lorenz.LorenzConfig attribute)
num_parameters (trphysx.embedding.embedding_model.EmbeddingModel attribute)
O
overwrite_cache (trphysx.config.args.DataArguments attribute)
P
parse() (trphysx.embedding.training.enn_args.EmbeddingParser method)
parse_args_into_dataclasses() (trphysx.config.arg_parser.HfArgumentParser method)
PhysConfig (class in trphysx.config.configuration_phys)
PhysformerBase (class in trphysx.transformer.phys_transformer_base)
PhysformerGPT2 (class in trphysx.transformer.phys_transformer_gpt2)
PhysformerTrain (class in trphysx.transformer.phys_transformer_helpers)
PhysicalDataset (class in trphysx.data_utils.dataset_phys)
plot_dir (trphysx.config.args.TrainingArguments attribute)
plot_max (trphysx.config.args.TrainingArguments attribute)
plotEmbeddingPrediction() (trphysx.viz.viz_cylinder.CylinderViz method)
(trphysx.viz.viz_model.Viz method)
plotMultiPrediction() (trphysx.viz.viz_lorenz.LorenzViz method)
plotPrediction() (trphysx.viz.viz_cylinder.CylinderViz method)
(trphysx.viz.viz_grayscott.GrayScottViz method)
(trphysx.viz.viz_lorenz.LorenzViz method)
(trphysx.viz.viz_model.Viz method)
plotPredictionScatter() (trphysx.viz.viz_lorenz.LorenzViz method)
plotPredictionVorticity() (trphysx.viz.viz_cylinder.CylinderViz method)
prepare_inputs_for_generation() (trphysx.transformer.generate_utils.GenerationMixin method)
push() (trphysx.utils.metrics.Metrics method)
R
read_cache() (trphysx.data_utils.dataset_phys.PhysicalDataset method)
recover() (trphysx.data_utils.dataset_grayscott.GrayscottPredictDataset method)
(trphysx.embedding.embedding_cylinder.CylinderEmbedding method)
(trphysx.embedding.embedding_grayscott.GrayScottEmbedding method)
(trphysx.embedding.embedding_lorenz.LorenzEmbedding method)
(trphysx.embedding.embedding_model.EmbeddingModel method)
S
save_model() (trphysx.embedding.embedding_model.EmbeddingModel method)
(trphysx.embedding.embedding_model.EmbeddingTrainingHead method)
(trphysx.transformer.phys_transformer_base.PhysformerBase method)
(trphysx.transformer.phys_transformer_helpers.PhysformerTrain method)
save_pretrained() (trphysx.config.configuration_phys.PhysConfig method)
save_steps (trphysx.config.args.TrainingArguments attribute)
saveFigure() (trphysx.viz.viz_model.Viz method)
seed (trphysx.config.args.TrainingArguments attribute)
set_input_embeddings() (trphysx.transformer.phys_transformer_base.PhysformerBase method)
set_seed() (in module trphysx.embedding.training.enn_trainer)
(in module trphysx.utils.trainer)
split_heads() (trphysx.transformer.attention.MaskedAttention method)
std (trphysx.embedding.training.enn_data_handler.EmbeddingDataHandler attribute)
stride (trphysx.config.args.DataArguments attribute)
T
tie_weights() (trphysx.transformer.phys_transformer_base.PhysformerBase method)
to_dict() (trphysx.config.configuration_phys.PhysConfig method)
to_json_file() (trphysx.config.configuration_phys.PhysConfig method)
to_json_string() (trphysx.config.configuration_phys.PhysConfig method)
train() (trphysx.embedding.training.enn_trainer.EmbeddingTrainer method)
(trphysx.utils.trainer.Trainer method)
train_batch_size (trphysx.config.args.TrainingArguments attribute)
Trainer (class in trphysx.utils.trainer)
training_h5_file (trphysx.config.args.DataArguments attribute)
training_step() (trphysx.utils.trainer.Trainer method)
TrainingArguments (class in trphysx.config.args)
transformer_file_or_path (trphysx.config.args.ModelArguments attribute)
trphysx.config.arg_parser (module)
trphysx.config.args (module)
trphysx.config.configuration_auto (module)
trphysx.config.configuration_cylinder (module)
trphysx.config.configuration_grayscott (module)
trphysx.config.configuration_lorenz (module)
trphysx.config.configuration_phys (module)
trphysx.data_utils.data_utils (module)
trphysx.data_utils.dataset_auto (module)
trphysx.data_utils.dataset_cylinder (module)
trphysx.data_utils.dataset_grayscott (module)
trphysx.data_utils.dataset_lorenz (module)
trphysx.data_utils.dataset_phys (module)
trphysx.embedding.embedding_auto (module)
trphysx.embedding.embedding_cylinder (module)
trphysx.embedding.embedding_grayscott (module)
trphysx.embedding.embedding_lorenz (module)
trphysx.embedding.embedding_model (module)
trphysx.embedding.training.enn_args (module)
trphysx.embedding.training.enn_data_handler (module)
trphysx.embedding.training.enn_trainer (module)
trphysx.transformer.attention (module)
trphysx.transformer.generate_utils (module)
trphysx.transformer.phys_transformer_base (module)
trphysx.transformer.phys_transformer_gpt2 (module)
trphysx.transformer.phys_transformer_helpers (module)
trphysx.transformer.utils (module)
trphysx.utils.metrics (module)
trphysx.utils.trainer (module)
trphysx.viz.viz_auto (module)
trphysx.viz.viz_cylinder (module)
trphysx.viz.viz_grayscott (module)
trphysx.viz.viz_lorenz (module)
trphysx.viz.viz_model (module)
U
update() (trphysx.config.configuration_phys.PhysConfig method)
V
Viz (class in trphysx.viz.viz_model)
viz_name (trphysx.config.args.ModelArguments attribute)
W
write_cache() (trphysx.data_utils.dataset_phys.PhysicalDataset method)
writeToHDF5() (trphysx.utils.metrics.Metrics method)
Read the Docs
v: latest
Versions
latest
stable
Downloads
pdf
html
On Read the Docs
Project Home
Builds