ertk.pytorch.models.SimpleClassificationModel
=============================================

.. currentmodule:: ertk.pytorch.models

.. autoclass:: SimpleClassificationModel
    :members:
    :show-inheritance:

    
    .. automethod:: __init__

    
    .. rubric:: Methods

    .. autosummary::
    
        ~SimpleClassificationModel.__init__
        ~SimpleClassificationModel.log_acc
        ~SimpleClassificationModel.training_step
        ~SimpleClassificationModel.validation_step

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

    
    
    .. rubric:: Attributes

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