https://mooseframework.inl.gov
Moose::LibtorchArtificialNeuralNetTrainer< SamplerType > Member List

This is the complete list of members for Moose::LibtorchArtificialNeuralNetTrainer< SamplerType >, including all inherited members.

_communicatorlibMesh::ParallelObjectprotected
_nnMoose::LibtorchArtificialNeuralNetTrainer< SamplerType >protected
comm() constlibMesh::ParallelObject
computeBatchSize(const unsigned int num_samples, const unsigned int num_batches)Moose::LibtorchArtificialNeuralNetTrainer< SamplerType >static
computeLocalBatchSize(const unsigned int batch_size, const unsigned int num_ranks)Moose::LibtorchArtificialNeuralNetTrainer< SamplerType >static
createOptimizer(const LibtorchArtificialNeuralNet &nn, const LibtorchTrainingOptions &options)Moose::LibtorchArtificialNeuralNetTrainer< SamplerType >static
LibtorchArtificialNeuralNetTrainer(LibtorchArtificialNeuralNet &nn, const Parallel::Communicator &comm)Moose::LibtorchArtificialNeuralNetTrainer< SamplerType >
n_processors() constlibMesh::ParallelObject
operator=(const ParallelObject &libmesh_dbg_var(other))libMesh::ParallelObject
ParallelObject(const Parallel::Communicator &comm_in)libMesh::ParallelObject
ParallelObject(const ParallelObject &other)libMesh::ParallelObject
processor_id() constlibMesh::ParallelObject
train(LibtorchDataset &dataset, const LibtorchTrainingOptions &options)Moose::LibtorchArtificialNeuralNetTrainer< SamplerType >virtual
~ParallelObject()=defaultlibMesh::ParallelObjectvirtual