ertk.pytorch.models.SimpleModel
===============================

.. currentmodule:: ertk.pytorch.models

.. autoclass:: SimpleModel
    :members:
    :show-inheritance:

    
    .. automethod:: __init__

    
    .. rubric:: Methods

    .. autosummary::
    
        ~SimpleModel.__init__
        ~SimpleModel.configure_optimizers
        ~SimpleModel.get_outputs_for_batch
        ~SimpleModel.log_loss
        ~SimpleModel.predict_step
        ~SimpleModel.training_step
        ~SimpleModel.validation_step

    
    .. rubric:: Inherited Methods
    .. autosummary::
    
        ~SimpleModel.add_module
        ~SimpleModel.all_gather
        ~SimpleModel.apply
        ~SimpleModel.backward
        ~SimpleModel.bfloat16
        ~SimpleModel.buffers
        ~SimpleModel.children
        ~SimpleModel.clip_gradients
        ~SimpleModel.configure_callbacks
        ~SimpleModel.configure_gradient_clipping
        ~SimpleModel.configure_model
        ~SimpleModel.configure_sharded_model
        ~SimpleModel.cpu
        ~SimpleModel.cuda
        ~SimpleModel.double
        ~SimpleModel.eval
        ~SimpleModel.extra_repr
        ~SimpleModel.float
        ~SimpleModel.forward
        ~SimpleModel.freeze
        ~SimpleModel.get_buffer
        ~SimpleModel.get_config_type
        ~SimpleModel.get_default_config
        ~SimpleModel.get_extra_state
        ~SimpleModel.get_model_class
        ~SimpleModel.get_parameter
        ~SimpleModel.get_submodule
        ~SimpleModel.half
        ~SimpleModel.ipu
        ~SimpleModel.load_from_checkpoint
        ~SimpleModel.load_state_dict
        ~SimpleModel.log
        ~SimpleModel.log_dict
        ~SimpleModel.lr_scheduler_step
        ~SimpleModel.lr_schedulers
        ~SimpleModel.make_model
        ~SimpleModel.manual_backward
        ~SimpleModel.modules
        ~SimpleModel.named_buffers
        ~SimpleModel.named_children
        ~SimpleModel.named_modules
        ~SimpleModel.named_parameters
        ~SimpleModel.on_after_backward
        ~SimpleModel.on_after_batch_transfer
        ~SimpleModel.on_before_backward
        ~SimpleModel.on_before_batch_transfer
        ~SimpleModel.on_before_optimizer_step
        ~SimpleModel.on_before_zero_grad
        ~SimpleModel.on_fit_end
        ~SimpleModel.on_fit_start
        ~SimpleModel.on_load_checkpoint
        ~SimpleModel.on_predict_batch_end
        ~SimpleModel.on_predict_batch_start
        ~SimpleModel.on_predict_end
        ~SimpleModel.on_predict_epoch_end
        ~SimpleModel.on_predict_epoch_start
        ~SimpleModel.on_predict_model_eval
        ~SimpleModel.on_predict_start
        ~SimpleModel.on_save_checkpoint
        ~SimpleModel.on_test_batch_end
        ~SimpleModel.on_test_batch_start
        ~SimpleModel.on_test_end
        ~SimpleModel.on_test_epoch_end
        ~SimpleModel.on_test_epoch_start
        ~SimpleModel.on_test_model_eval
        ~SimpleModel.on_test_model_train
        ~SimpleModel.on_test_start
        ~SimpleModel.on_train_batch_end
        ~SimpleModel.on_train_batch_start
        ~SimpleModel.on_train_end
        ~SimpleModel.on_train_epoch_end
        ~SimpleModel.on_train_epoch_start
        ~SimpleModel.on_train_start
        ~SimpleModel.on_validation_batch_end
        ~SimpleModel.on_validation_batch_start
        ~SimpleModel.on_validation_end
        ~SimpleModel.on_validation_epoch_end
        ~SimpleModel.on_validation_epoch_start
        ~SimpleModel.on_validation_model_eval
        ~SimpleModel.on_validation_model_train
        ~SimpleModel.on_validation_model_zero_grad
        ~SimpleModel.on_validation_start
        ~SimpleModel.optimizer_step
        ~SimpleModel.optimizer_zero_grad
        ~SimpleModel.optimizers
        ~SimpleModel.parameters
        ~SimpleModel.predict_dataloader
        ~SimpleModel.prepare_data
        ~SimpleModel.preprocess_input
        ~SimpleModel.print
        ~SimpleModel.register_backward_hook
        ~SimpleModel.register_buffer
        ~SimpleModel.register_forward_hook
        ~SimpleModel.register_forward_pre_hook
        ~SimpleModel.register_full_backward_hook
        ~SimpleModel.register_full_backward_pre_hook
        ~SimpleModel.register_load_state_dict_post_hook
        ~SimpleModel.register_module
        ~SimpleModel.register_parameter
        ~SimpleModel.register_state_dict_pre_hook
        ~SimpleModel.requires_grad_
        ~SimpleModel.save_hyperparameters
        ~SimpleModel.set_extra_state
        ~SimpleModel.setup
        ~SimpleModel.share_memory
        ~SimpleModel.state_dict
        ~SimpleModel.teardown
        ~SimpleModel.test_dataloader
        ~SimpleModel.test_step
        ~SimpleModel.to
        ~SimpleModel.to_empty
        ~SimpleModel.to_onnx
        ~SimpleModel.to_torchscript
        ~SimpleModel.toggle_optimizer
        ~SimpleModel.train
        ~SimpleModel.train_dataloader
        ~SimpleModel.transfer_batch_to_device
        ~SimpleModel.type
        ~SimpleModel.unfreeze
        ~SimpleModel.untoggle_optimizer
        ~SimpleModel.val_dataloader
        ~SimpleModel.valid_models
        ~SimpleModel.xpu
        ~SimpleModel.zero_grad
    
    
    

    
    
    .. rubric:: Attributes

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