torch_model_is_training Function

public function torch_model_is_training(self) result(is_training)

Determines whether a model is set up for training

Type Bound

torch_model

Arguments

Type IntentOptional Attributes Name
class(torch_model), intent(in) :: self

Model to query

Return Value logical

Whether the model is set up for training


Source Code

  function torch_model_is_training(self) result(is_training)
    class(torch_model), intent(in) :: self  !! Model to query
    logical :: is_training                  !! Whether the model is set up for training

    interface
      function torch_jit_model_is_training_c(model_c) result(is_training_c) &
          bind(c, name = 'torch_jit_module_is_training')
        use, intrinsic :: iso_c_binding, only : c_bool, c_ptr
        implicit none
        type(c_ptr), value, intent(in) :: model_c
        logical(c_bool) :: is_training_c
      end function torch_jit_model_is_training_c
    end interface

    is_training = torch_jit_model_is_training_c(self%p)
  end function torch_model_is_training