-
Notifications
You must be signed in to change notification settings - Fork 18
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
0 parents
commit 1b18c9a
Showing
62 changed files
with
24,099 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
include: ["_sources", "_static"] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,14 @@ | ||
.. _api-label: | ||
|
||
TorchFort API | ||
============= | ||
|
||
The following sections describe the types and functions available in the TorchFort library for C/C++ and Fortran programs and | ||
the also the configuration file structure and available options. | ||
|
||
.. toctree:: | ||
|
||
api/config | ||
api/c_api | ||
api/f_api | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,277 @@ | ||
.. _torchfort_api_c-ref: | ||
|
||
############### | ||
TorchFort C API | ||
############### | ||
|
||
These are all the types and functions available in the TorchFort C API. | ||
|
||
******* | ||
General | ||
******* | ||
|
||
Types | ||
----- | ||
|
||
.. _torchfort_datatype_t-ref: | ||
|
||
torchfort_datatype_t | ||
____________________ | ||
.. doxygenenum :: torchfort_datatype_t | ||
------ | ||
|
||
.. _torchfort_result_t-ref: | ||
|
||
torchfort_result_t | ||
__________________ | ||
.. doxygenenum :: torchfort_result_t | ||
------ | ||
|
||
Global Context Settings | ||
------------------------ | ||
|
||
These are global routines which affect the behavior of the libtorch backend. It is therefore recommended to call these functions before any other TorchFort calls are made. | ||
|
||
.. _torchfort_set_cudnn_benchmark-ref: | ||
|
||
torchfort_set_cudnn_benchmark | ||
_____________________________ | ||
.. doxygenfunction:: torchfort_set_cudnn_benchmark | ||
|
||
|
||
.. _torchfort_general_c-ref: | ||
|
||
******************* | ||
Supervised Learning | ||
******************* | ||
|
||
Model Creation | ||
----------------------------------- | ||
|
||
.. _torchfort_create_model-ref: | ||
|
||
torchfort_create_model | ||
______________________ | ||
.. doxygenfunction:: torchfort_create_model | ||
|
||
------ | ||
|
||
.. _torchfort_create_distributed-model-ref: | ||
|
||
torchfort_create_distributed_model | ||
__________________________________ | ||
.. doxygenfunction:: torchfort_create_distributed_model | ||
|
||
------ | ||
|
||
Model Training/Inference | ||
----------------------------------- | ||
|
||
.. _torchfort_train-ref: | ||
|
||
torchfort_train | ||
_______________ | ||
.. doxygenfunction:: torchfort_train | ||
|
||
------ | ||
|
||
.. _torchfort_inference-ref: | ||
|
||
torchfort_inference | ||
___________________ | ||
.. doxygenfunction:: torchfort_inference | ||
|
||
------ | ||
|
||
Model Management | ||
---------------- | ||
|
||
.. _torchfort_save_model-ref: | ||
|
||
torchfort_save_model | ||
____________________ | ||
.. doxygenfunction:: torchfort_save_model | ||
|
||
------ | ||
|
||
.. _torchfort_load_model-ref: | ||
|
||
torchfort_load_model | ||
____________________ | ||
.. doxygenfunction:: torchfort_load_model | ||
|
||
------ | ||
|
||
.. _torchfort_save_checkpoint-ref: | ||
|
||
torchfort_save_checkpoint | ||
_________________________ | ||
.. doxygenfunction:: torchfort_save_checkpoint | ||
|
||
------ | ||
|
||
.. _torchfort_load_checkpoint-ref: | ||
|
||
torchfort_load_checkpoint | ||
_________________________ | ||
.. doxygenfunction:: torchfort_load_checkpoint | ||
|
||
------ | ||
|
||
Weights and Biases Logging | ||
-------------------------- | ||
|
||
.. _torchfort_wandb_log_int-ref: | ||
|
||
torchfort_wandb_log_int | ||
_______________________ | ||
.. doxygenfunction:: torchfort_wandb_log_int | ||
|
||
------ | ||
|
||
.. _torchfort_wandb_log_float-ref: | ||
|
||
torchfort_wandb_log_float | ||
_________________________ | ||
.. doxygenfunction:: torchfort_wandb_log_float | ||
|
||
------ | ||
|
||
.. _torchfort_wandb_log_double-ref: | ||
|
||
torchfort_wandb_log_double | ||
__________________________ | ||
.. doxygenfunction:: torchfort_wandb_log_double | ||
|
||
------ | ||
|
||
.. _torchfort_rl_c-ref: | ||
|
||
********************** | ||
Reinforcement Learning | ||
********************** | ||
|
||
System Creation | ||
----------------------------------- | ||
|
||
Basic routines to create and register a reinforcement learning system in the internal registry. A (synchronous) data parallel distributed option is available. | ||
|
||
.. _torchfort_rl_off_policy_create_system-ref: | ||
|
||
torchfort_rl_off_policy_create_system | ||
_____________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_create_system | ||
|
||
------ | ||
|
||
.. _torchfort_rl_off_policy_create_distributed_system-ref: | ||
|
||
torchfort_rl_off_policy_create_distributed_system | ||
_________________________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_create_distributed_system | ||
|
||
------ | ||
|
||
Training/Evaluation | ||
----------------------------------------- | ||
|
||
These routines are used for training the reinforcement learning system or for steering the environment. | ||
|
||
.. _torchfort_rl_off_policy_train_step-ref: | ||
|
||
torchfort_rl_off_policy_train_step | ||
__________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_train_step | ||
|
||
------ | ||
|
||
.. _torchfort_rl_off_policy_predict_explore-ref: | ||
|
||
torchfort_rl_off_policy_predict_explore | ||
_______________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_predict_explore | ||
|
||
------ | ||
|
||
.. _torchfort_rl_off_policy_predict-ref: | ||
|
||
torchfort_rl_off_policy_predict | ||
_______________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_predict | ||
|
||
------ | ||
|
||
.. _torchfort_rl_off_policy_evaluate-ref: | ||
|
||
torchfort_rl_off_policy_evaluate | ||
________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_evaluate | ||
|
||
------ | ||
|
||
System Management | ||
----------------- | ||
|
||
The purpose of these routines is to manage the reinforcement learning systems internal data. It allows the user to add tuples to the replay buffer and query the system for readiness. Additionally, save and restore functionality is also provided. | ||
|
||
.. _torchfort_rl_off_policy_update_replay_buffer-ref: | ||
|
||
torchfort_rl_off_policy_update_replay_buffer | ||
____________________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_update_replay_buffer | ||
|
||
------ | ||
|
||
.. _torchfort_rl_off_policy_is_ready-ref: | ||
|
||
torchfort_rl_off_policy_is_ready | ||
________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_is_ready | ||
|
||
------ | ||
|
||
.. _torchfort_rl_off_policy_save_checkpoint-ref: | ||
|
||
torchfort_rl_off_policy_save_checkpoint | ||
_______________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_save_checkpoint | ||
|
||
------ | ||
|
||
.. _torchfort_rl_off_policy_load_checkpoint-ref: | ||
|
||
torchfort_rl_off_policy_load_checkpoint | ||
_______________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_load_checkpoint | ||
|
||
------ | ||
|
||
|
||
Weights and Biases Logging | ||
-------------------------- | ||
|
||
The reinforcement learning system performs logging for all involved networks automatically during training. The following routines are provided for additional logging of system relevant quantities, such as e.g. the accumulated reward. | ||
|
||
.. _torchfort_rl_off_policy_wandb_log_int-ref: | ||
|
||
torchfort_rl_off_policy_wandb_log_int | ||
_____________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_wandb_log_int | ||
|
||
------ | ||
|
||
.. _torchfort_rl_off_policy_wandb_log_float-ref: | ||
|
||
torchfort_rl_off_policy_wandb_log_float | ||
_______________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_wandb_log_float | ||
|
||
------ | ||
|
||
.. _torchfort_rl_off_policy_wandb_log_double-ref: | ||
|
||
torchfort_rl_off_policy_wandb_log_double | ||
________________________________________ | ||
.. doxygenfunction:: torchfort_rl_off_policy_wandb_log_double |
Oops, something went wrong.