Skip to main content

Ctrl+K

Site Navigation

Get Started
Use Cases
Example Gallery
Library
Docs
Resources

Try Ray on Anyscale

Site Navigation

Get Started
Use Cases
Example Gallery
Library
Docs
Resources

Try Ray on Anyscale

Overview
Getting Started
Installation
Use Cases
- Ray for ML Infrastructure
Example Gallery
Ecosystem
Ray Core
Ray Data
Ray Train
Ray Tune
Ray Serve
Ray RLlib
More Libraries
Ray Clusters
Monitoring and Debugging
Developer Guides
Glossary
Security

RLlib: Industry-Grade, Scalable Reinforcement Learning
ray.rllib.models.torch.torch_modelv2.TorchModelV2
ray.rllib.mo...

ray.rllib.models.torch.torch_modelv2.TorchModelV2.context#

TorchModelV2.context() → AbstractContextManager#: Returns a contextmanager for the current forward pass.

previous

ray.rllib.models.torch.torch_modelv2.TorchModelV2.__init__

next

ray.rllib.models.torch.torch_modelv2.TorchModelV2.custom_loss

On this page

TorchModelV2.context()

Thanks for the feedback!

Was this helpful?

Yes

No

Feedback

Submit

© Copyright 2024, The Ray Team.

Created using Sphinx 7.3.7.

Built with the PyData Sphinx Theme 0.14.1.