torch\_concepts.nn.ConceptBottleneckModel\_Joint
================================================

.. currentmodule:: torch_concepts.nn

.. autoclass:: ConceptBottleneckModel_Joint

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~ConceptBottleneckModel_Joint.CHECKPOINT_HYPER_PARAMS_KEY
      ~ConceptBottleneckModel_Joint.CHECKPOINT_HYPER_PARAMS_NAME
      ~ConceptBottleneckModel_Joint.CHECKPOINT_HYPER_PARAMS_TYPE
      ~ConceptBottleneckModel_Joint.T_destination
      ~ConceptBottleneckModel_Joint.automatic_optimization
      ~ConceptBottleneckModel_Joint.backbone
      ~ConceptBottleneckModel_Joint.call_super_init
      ~ConceptBottleneckModel_Joint.current_epoch
      ~ConceptBottleneckModel_Joint.device
      ~ConceptBottleneckModel_Joint.device_mesh
      ~ConceptBottleneckModel_Joint.dtype
      ~ConceptBottleneckModel_Joint.dump_patches
      ~ConceptBottleneckModel_Joint.example_input_array
      ~ConceptBottleneckModel_Joint.fabric
      ~ConceptBottleneckModel_Joint.global_rank
      ~ConceptBottleneckModel_Joint.global_step
      ~ConceptBottleneckModel_Joint.hparams
      ~ConceptBottleneckModel_Joint.hparams_initial
      ~ConceptBottleneckModel_Joint.latent_encoder
      ~ConceptBottleneckModel_Joint.local_rank
      ~ConceptBottleneckModel_Joint.logger
      ~ConceptBottleneckModel_Joint.loggers
      ~ConceptBottleneckModel_Joint.on_gpu
      ~ConceptBottleneckModel_Joint.strict_loading
      ~ConceptBottleneckModel_Joint.trainer
      ~ConceptBottleneckModel_Joint.training