Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
AMS documentation
AMS documentation
  • Installation
  • Full API
    • Namespace @0
    • Namespace @1
    • Namespace @36
    • Namespace abc
    • Namespace action
    • Namespace ams
    • Namespace ams::db
    • Namespace ams::internal
    • Namespace ams::ml
    • Namespace ams::tensor
    • Namespace ams_flux
    • Namespace ams_jobs
    • Namespace AMSBroker
    • Namespace AMSDBStage
    • Namespace AMSDeploy
    • Namespace AMSOrchestrator
    • Namespace AMSStore
    • Namespace AMSTrain
    • Namespace enum
    • Namespace faccessors
    • Namespace fmt
    • Namespace loader
    • Namespace monitor
    • Namespace nlohmann
    • Namespace orchestrator
    • Namespace rmq
    • Namespace setup
    • Namespace stage
    • Namespace std
    • Namespace store
    • Namespace store_types
    • Namespace util
    • Namespace views
    • Namespace wf_manager
    • Struct AMSAllocator
    • Struct AMSDefaultDeviceAllocator
    • Struct AMSDefaultHostAllocator
    • Struct AMSDefaultPinnedAllocator
    • Template Struct CalculateSmallVectorDefaultInlinedElements
    • Struct EvalContext
    • Struct IndexMap
    • Struct IndexMap::FieldInfo
    • Struct LogConfig
    • Template Struct BaseModel::dependent_false
    • Template Struct SmallVectorAlignmentAndSize
    • Template Struct SmallVectorStorage
    • Template Struct SmallVectorStorage< T, 0 >
    • Struct TensorBundle
    • Struct TensorBundle::Item
    • Template Struct formatter< ams::AMSResourceType >
    • Template Struct formatter< std::filesystem::path >
    • Class UserAction
    • Class Action
    • Class AMSError
    • Class AMSTensor
    • Class AMSWorkflow
    • Template Class ArrayRef
    • Class BaseDB
    • Class DBManager
    • Class FileDB
    • Class FilesystemInterface
    • Class RMQInterface
    • Template Class iterator_range
    • Class LayoutTransform
    • Class Logger
    • Class AbstractModel
    • Class BaseModel
    • Class InferenceModel
    • Template Class MutableArrayRef
    • Template Class OwningArrayRef
    • Class Pipeline
    • Class PointwiseConcatTransform
    • Class Policy
    • Class ResourceManager
    • Template Class SmallVector
    • Template Class SmallVectorBase
    • Template Class SmallVectorImpl
    • Template Class SmallVectorTemplateBase
    • Template Class SmallVectorTemplateBase< T, true >
    • Template Class SmallVectorTemplateCommon
    • Class _WorkItem
    • Class AMSFakeFluxOrchestatorExecutor
    • Class AMSFluxExecutor
    • Class AMSFluxExecutorFuture
    • Class AMSFluxOrchestratorExecutor
    • Class AMSDomainJob
    • Class AMSFSStageJob
    • Class AMSFSTempStageJob
    • Class AMSJob
    • Class AMSJobResources
    • Class AMSMLJob
    • Class AMSMLTrainJob
    • Class AMSNetworkStageJob
    • Class AMSOrchestratorJob
    • Class AMSStageJob
    • Class AMSSubSelectJob
    • Class AddToStore
    • Class CreateStore
    • Class RemoveFromStore
    • Class SearchStore
    • Class StoreCommand
    • Class FileReader
    • Class FileWriter
    • Class HDF5CLibReader
    • Class HDF5PackedReader
    • Class HDF5PackedWriter
    • Class HDF5Writer
    • Template Class isDouble
    • Template Class isDouble< double >
    • Template Class isDouble< float >
    • Class AMSMonitor
    • Class AMSFakeRMQUpdate
    • Class AMSJobReceiverStage
    • Class AMSRMQMessagePrinter
    • Class AvailableDomains
    • Class DomainSpec
    • Class DomainSpec::TrackJobStats
    • Class JobState
    • Class RequestProcessor
    • Class RMQStatusUpdate
    • Class StatusPrinter
    • Class TrainJobScheduler
    • Class AMSChannel
    • Class AMSFanOutProducer
    • Class AMSMessage
    • Class AMSRMQConfiguration
    • Class AMSSyncProducer
    • Class AsyncConsumer
    • Class AsyncFanOutConsumer
    • Class BlockingClient
    • Class StatusPoller
    • Class AMSShutdown
    • Class DataBlob
    • Class ForwardTask
    • Class FSLoaderTask
    • Class FSPipeline
    • Class FSWriteTask
    • Class MessageType
    • Class Pipeline
    • Class PushToStore
    • Class QueueMessage
    • Class RMQControlMessageTask
    • Class RMQDomainDataLoaderTask
    • Class RMQPipeline
    • Class Task
    • Class AMSDataStore
    • Class Entry
    • Class EntryType
    • Class AMSModelDescr
    • Class UQAggregate
    • Class UQType
    • Class SurrogateModel
    • Class AMSDataView
    • Class AMSHDF5VirtualDBReader
    • Class AMSHDF5VirtualDBReader::DataDescr
    • Class AMSWorkflowManager
    • Class JobList
    • Class Partition
    • Enum AMSDBType
    • Enum AMSDType
    • Enum AMSErrorType
    • Enum AMSExecPolicy
    • Enum AMSResourceType
    • Enum LogLevel
    • Function ams::AMSCExecute
    • Function ams::AMSConfigureFSDatabase
    • Function ams::AMSCreateExecutor
    • Function ams::AMSDestroyExecutor
    • Function ams::AMSExecute
    • Function ams::AMSFinalize
    • Function ams::AMSGetAllocatorName
    • Function ams::AMSGetDatabaseName
    • Function ams::AMSInit
    • Function ams::AMSQueryModel
    • Function ams::AMSRegisterAbstractModel
    • Function ams::AMSSetAllocator
    • Template Function ams::capacity_in_bytes
    • Function ams::db::getDBType
    • Function ams::db::getDBTypeAsStr
    • Function ams::detectRank
    • Function ams::internal::_get_allocator
    • Function ams::internal::_raw_copy
    • Function ams::internal::_release_allocator
    • Template Function ams::MakeError
    • Function ams::ml::parseName
    • Function ams::ml::parsePath
    • Template Function ams::operator!=(ArrayRef<T>, ArrayRef<T>)
    • Template Function ams::operator!=(SmallVectorImpl<T>&, ArrayRef<T>)
    • Function ams::operator<<(std::ostream&, const AMSError&)
    • Template Function ams::operator<<(std::ostream&, const SmallVector<T, N>&)
    • Template Function ams::operator==(ArrayRef<T>, ArrayRef<T>)
    • Template Function ams::operator==(SmallVectorImpl<T>&, ArrayRef<T>)
    • Template Function ams::roundUp
    • Function ams::style_for
    • Function ams::tensor::maskTensor
    • Template Function ams::to_vector(R&&)
    • Template Function ams::to_vector(R&&)
    • Template Function ams::to_vector_of(R&&)
    • Template Function ams::to_vector_of(R&&)
    • Function ams_jobs::constuct_cli_cmd
    • Function ams_jobs::get_echo_job
    • Function ams_jobs::nested_instance_job_descr
    • Function AMSBroker::main
    • Function AMSDBStage::main
    • Function amsDebug
    • Function AMSDeploy::get_partition_uri
    • Function AMSDeploy::main
    • Function AMSDeploy::verify_arg
    • Function AMSOrchestrator::main
    • Function AMSStore::main
    • Function amsToTorchDevice
    • Function amsToTorchDType
    • Function amsToTorchTensors
    • Function AMSTrain::main
    • Function callAMS(ams::AMSWorkflow *, DomainLambda, const ams::SmallVector<ams::AMSTensor>&, ams::SmallVector<ams::AMSTensor>&, ams::SmallVector<ams::AMSTensor>&)
    • Function callAMS(ams::AMSWorkflow *, ams::DomainLambda, const ams::SmallVector<ams::AMSTensor>&, ams::SmallVector<ams::AMSTensor>&, ams::SmallVector<ams::AMSTensor>&)
    • Function callApplication(ams::DomainLambda, ams::MutableArrayRef<torch::Tensor>, ams::MutableArrayRef<torch::Tensor>, ams::MutableArrayRef<torch::Tensor>)
    • Function callApplication(ams::DomainLambda, ams::MutableArrayRef<torch::Tensor>, ams::MutableArrayRef<torch::Tensor>, ams::MutableArrayRef<torch::Tensor>)
    • Template Function computeNumElements
    • Function dtype_to_size
    • Function dtypeToString
    • Function faccessors::get_reader
    • Function faccessors::get_writer
    • Function faccessors::main
    • Function getAMSDTypeAsString
    • Function getAMSResourceTypeAsString
    • Function getDTypeAsString
    • Template Function getNewCapacity
    • Template Function is_real_equal
    • Function loader::load_class
    • Function memUsage(double&, double&)
    • Function memUsage(double&, double&)
    • Function orchestrator::run
    • Function orchestrator::thread_safe_call
    • Function random_uq_host
    • Function report_at_maximum_capacity
    • Function report_size_overflow
    • Function rmq::default_ams_callback
    • Function shapeToString
    • Function SmallVectorToString
    • Function stage::get_pipeline
    • Template Function std::swap(ams::SmallVectorImpl<T>&, ams::SmallVectorImpl<T>&)
    • Template Function std::swap(ams::SmallVector<T, N>&, ams::SmallVector<T, N>&)
    • Function store::create_store_directories
    • Function tensorSizeToString
    • Function torchDeviceToAMSDevice
    • Function torchDTypeToAMSType
    • Function torchDTypeToHDF5Type
    • Function torchToAMSTensors
    • Function util::generate_tls_certificate
    • Function util::get_unique_fn
    • Function util::mkdir
    • Function wf_manager::get_allocation_resources
    • Variable stage::BATCH_SIZE
    • Variable store::Base
    • Define __AMS_FILE__
    • Define __AMS_LINE__
    • Define AMS_CDEBUG
    • Define AMS_CFATAL
    • Define AMS_CINFO
    • Define AMS_CWARNING
    • Define AMS_DBG
    • Define AMS_FATAL
    • Define AMS_INFO
    • Define AMS_MAKE_ERROR
    • Define AMS_MAKE_ERROR_OBJ
    • Define AMS_WARNING
    • Define AMSPRINT
    • Define CALIPER
    • Define GETNAME
    • Define GETNAME2
    • Define MPI_CALL
    • Define PERFFASPECT
    • Define THROW
    • Define UNDEFINED_FUNC
    • Typedef ams::AMSCAbstrModel
    • Typedef ams::AMSExecutor
    • Typedef ams::AMSExpected
    • Typedef ams::AMSStatus
    • Typedef ams::DomainCFn
    • Typedef ams::DomainLambda
    • Typedef ams::EnableIfConvertibleToInputIterator
    • Typedef ams::SmallVectorSizeType
    • Typedef ams::ValueTypeFromRangeType
    • Typedef enable_if_t
    • Typedef MPI_Comm
Back to top
View this page

Function AMSTrain::main¶

  • Defined in File AMSTrain.py

Function Documentation¶

AMSTrain.main()¶
Next
Function callAMS(ams::AMSWorkflow *, DomainLambda, const ams::SmallVector<ams::AMSTensor>&, ams::SmallVector<ams::AMSTensor>&, ams::SmallVector<ams::AMSTensor>&)
Previous
Function amsToTorchTensors
Copyright © 2023-2025, Lawrence Livermore National Security, LLC
Made with Sphinx and @pradyunsg's Furo
On this page
  • Function AMSTrain::main
    • Function Documentation
      • main()