Skip to content

Commit

Permalink
Updates to README main
Browse files Browse the repository at this point in the history
  • Loading branch information
jsschreck committed Aug 21, 2024
1 parent c87c486 commit e29ab8c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ which is the sum of aleatoric and epistemic contributions, respectively. The MIL
For detailed information about training with Keras, refer to [the Keras training details README](docs/source/keras.md). There three scripts for training three regression models, and one for training categorical models. The regression examples are trained on our surface layer ("SL") dataset for predicting latent heat and other quantities,
and the categorical example is trained on a precipitation dataset ("p-type").

For pyTorch, please visit the [the pyTorch training details README](docs/source/torch.md). There is one training script that works for both evidential and standard classification tasks. Torch examples use the same datasets as the Keras models. The torch code can be used with DDP and FSDP.
For pyTorch, please visit the [the pyTorch training details README](docs/source/torch.md) where details on training scripts for both evidential standard classification tasks are detailed. Torch examples use the same datasets as the Keras models. The torch training code will also scale on GPUs, and is compatitible with DDP and FSDP.

<!--
### 1a. Train/evaluate a deterministic multi-layer perceptrion (MLP) on the SL dataset:
Expand Down

0 comments on commit e29ab8c

Please sign in to comment.