tfep.app.base.TFEPMapBase
=========================

.. currentmodule:: tfep.app.base

.. autoclass:: TFEPMapBase
   :members:
   :show-inheritance:
   :inherited-members: Module,LightningModule

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~TFEPMapBase.__init__
      ~TFEPMapBase.add_module
      ~TFEPMapBase.all_gather
      ~TFEPMapBase.apply
      ~TFEPMapBase.are_axes_atoms_mapped
      ~TFEPMapBase.backward
      ~TFEPMapBase.bfloat16
      ~TFEPMapBase.buffers
      ~TFEPMapBase.children
      ~TFEPMapBase.clip_gradients
      ~TFEPMapBase.compile
      ~TFEPMapBase.configure_callbacks
      ~TFEPMapBase.configure_flow
      ~TFEPMapBase.configure_gradient_clipping
      ~TFEPMapBase.configure_model
      ~TFEPMapBase.configure_optimizers
      ~TFEPMapBase.configure_sharded_model
      ~TFEPMapBase.cpu
      ~TFEPMapBase.create_dataset
      ~TFEPMapBase.create_partial_flow
      ~TFEPMapBase.create_universe
      ~TFEPMapBase.cuda
      ~TFEPMapBase.determine_atom_indices
      ~TFEPMapBase.double
      ~TFEPMapBase.eval
      ~TFEPMapBase.extra_repr
      ~TFEPMapBase.float
      ~TFEPMapBase.forward
      ~TFEPMapBase.freeze
      ~TFEPMapBase.get_buffer
      ~TFEPMapBase.get_conditioning_indices
      ~TFEPMapBase.get_extra_state
      ~TFEPMapBase.get_mapped_indices
      ~TFEPMapBase.get_nonfixed_indices
      ~TFEPMapBase.get_parameter
      ~TFEPMapBase.get_reference_atoms_indices
      ~TFEPMapBase.get_submodule
      ~TFEPMapBase.half
      ~TFEPMapBase.inverse
      ~TFEPMapBase.ipu
      ~TFEPMapBase.load_from_checkpoint
      ~TFEPMapBase.load_state_dict
      ~TFEPMapBase.log
      ~TFEPMapBase.log_dict
      ~TFEPMapBase.lr_scheduler_step
      ~TFEPMapBase.lr_schedulers
      ~TFEPMapBase.manual_backward
      ~TFEPMapBase.modules
      ~TFEPMapBase.mtia
      ~TFEPMapBase.named_buffers
      ~TFEPMapBase.named_children
      ~TFEPMapBase.named_modules
      ~TFEPMapBase.named_parameters
      ~TFEPMapBase.on_after_backward
      ~TFEPMapBase.on_after_batch_transfer
      ~TFEPMapBase.on_before_backward
      ~TFEPMapBase.on_before_batch_transfer
      ~TFEPMapBase.on_before_optimizer_step
      ~TFEPMapBase.on_before_zero_grad
      ~TFEPMapBase.on_fit_end
      ~TFEPMapBase.on_fit_start
      ~TFEPMapBase.on_load_checkpoint
      ~TFEPMapBase.on_predict_batch_end
      ~TFEPMapBase.on_predict_batch_start
      ~TFEPMapBase.on_predict_end
      ~TFEPMapBase.on_predict_epoch_end
      ~TFEPMapBase.on_predict_epoch_start
      ~TFEPMapBase.on_predict_model_eval
      ~TFEPMapBase.on_predict_start
      ~TFEPMapBase.on_save_checkpoint
      ~TFEPMapBase.on_test_batch_end
      ~TFEPMapBase.on_test_batch_start
      ~TFEPMapBase.on_test_end
      ~TFEPMapBase.on_test_epoch_end
      ~TFEPMapBase.on_test_epoch_start
      ~TFEPMapBase.on_test_model_eval
      ~TFEPMapBase.on_test_model_train
      ~TFEPMapBase.on_test_start
      ~TFEPMapBase.on_train_batch_end
      ~TFEPMapBase.on_train_batch_start
      ~TFEPMapBase.on_train_end
      ~TFEPMapBase.on_train_epoch_end
      ~TFEPMapBase.on_train_epoch_start
      ~TFEPMapBase.on_train_start
      ~TFEPMapBase.on_validation_batch_end
      ~TFEPMapBase.on_validation_batch_start
      ~TFEPMapBase.on_validation_end
      ~TFEPMapBase.on_validation_epoch_end
      ~TFEPMapBase.on_validation_epoch_start
      ~TFEPMapBase.on_validation_model_eval
      ~TFEPMapBase.on_validation_model_train
      ~TFEPMapBase.on_validation_model_zero_grad
      ~TFEPMapBase.on_validation_start
      ~TFEPMapBase.optimizer_step
      ~TFEPMapBase.optimizer_zero_grad
      ~TFEPMapBase.optimizers
      ~TFEPMapBase.parameters
      ~TFEPMapBase.predict_dataloader
      ~TFEPMapBase.predict_step
      ~TFEPMapBase.prepare_data
      ~TFEPMapBase.print
      ~TFEPMapBase.register_backward_hook
      ~TFEPMapBase.register_buffer
      ~TFEPMapBase.register_forward_hook
      ~TFEPMapBase.register_forward_pre_hook
      ~TFEPMapBase.register_full_backward_hook
      ~TFEPMapBase.register_full_backward_pre_hook
      ~TFEPMapBase.register_load_state_dict_post_hook
      ~TFEPMapBase.register_load_state_dict_pre_hook
      ~TFEPMapBase.register_module
      ~TFEPMapBase.register_parameter
      ~TFEPMapBase.register_state_dict_post_hook
      ~TFEPMapBase.register_state_dict_pre_hook
      ~TFEPMapBase.requires_grad_
      ~TFEPMapBase.save_hyperparameters
      ~TFEPMapBase.set_extra_state
      ~TFEPMapBase.set_submodule
      ~TFEPMapBase.setup
      ~TFEPMapBase.share_memory
      ~TFEPMapBase.state_dict
      ~TFEPMapBase.teardown
      ~TFEPMapBase.test_dataloader
      ~TFEPMapBase.test_step
      ~TFEPMapBase.to
      ~TFEPMapBase.to_empty
      ~TFEPMapBase.to_onnx
      ~TFEPMapBase.to_torchscript
      ~TFEPMapBase.toggle_optimizer
      ~TFEPMapBase.train
      ~TFEPMapBase.train_dataloader
      ~TFEPMapBase.training_step
      ~TFEPMapBase.transfer_batch_to_device
      ~TFEPMapBase.type
      ~TFEPMapBase.unfreeze
      ~TFEPMapBase.untoggle_optimizer
      ~TFEPMapBase.val_dataloader
      ~TFEPMapBase.validation_step
      ~TFEPMapBase.xpu
      ~TFEPMapBase.zero_grad
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~TFEPMapBase.CHECKPOINT_HYPER_PARAMS_KEY
      ~TFEPMapBase.CHECKPOINT_HYPER_PARAMS_NAME
      ~TFEPMapBase.CHECKPOINT_HYPER_PARAMS_TYPE
      ~TFEPMapBase.T_destination
      ~TFEPMapBase.automatic_optimization
      ~TFEPMapBase.call_super_init
      ~TFEPMapBase.current_epoch
      ~TFEPMapBase.device
      ~TFEPMapBase.device_mesh
      ~TFEPMapBase.dtype
      ~TFEPMapBase.dump_patches
      ~TFEPMapBase.example_input_array
      ~TFEPMapBase.fabric
      ~TFEPMapBase.global_rank
      ~TFEPMapBase.global_step
      ~TFEPMapBase.hparams
      ~TFEPMapBase.hparams_initial
      ~TFEPMapBase.local_rank
      ~TFEPMapBase.logger
      ~TFEPMapBase.loggers
      ~TFEPMapBase.n_conditioning_atoms
      ~TFEPMapBase.n_conditioning_dofs
      ~TFEPMapBase.n_fixed_atoms
      ~TFEPMapBase.n_mapped_atoms
      ~TFEPMapBase.n_mapped_dofs
      ~TFEPMapBase.n_nonfixed_atoms
      ~TFEPMapBase.n_nonfixed_dofs
      ~TFEPMapBase.on_gpu
      ~TFEPMapBase.strict_loading
      ~TFEPMapBase.trainer
      ~TFEPMapBase.dataset
      ~TFEPMapBase.training