https://mooseframework.inl.gov
LibtorchNeuralNetControlTransfer.h
Go to the documentation of this file.
1 //* This file is part of the MOOSE framework
2 //* https://mooseframework.inl.gov
3 //*
4 //* All rights reserved, see COPYRIGHT for full restrictions
5 //* https://github.com/idaholab/moose/blob/master/COPYRIGHT
6 //*
7 //* Licensed under LGPL 2.1, please see LICENSE for details
8 //* https://www.gnu.org/licenses/lgpl-2.1.html
9 
10 #ifdef MOOSE_LIBTORCH_ENABLED
11 
12 #pragma once
13 
14 // torch-based includes
16 
17 #include "MultiAppTransfer.h"
19 
21 {
22 public:
24 
26 
27  virtual void execute() override;
28 
29 protected:
31  const std::string _control_name;
32 
35 };
36 
37 #endif
const LibtorchDRLControlTrainer & _trainer
The trainer object which will contains the control neural net.
const InputParameters & parameters() const
const std::string _control_name
The name of the control object on the other app where we want to copy our neural net.
Interface for objects that need to use samplers.
This trainer is responsible for training neural networks that efficiently control different processes...
LibtorchNeuralNetControlTransfer(const InputParameters &parameters)