wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-19 12:30:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

[RLlib] JAXPolicy prep PR #2 (move get_activation_fn (backward-compatibly), minor fixes and preparations). (#13091)

Browse Source
This commit is contained in:
Sven Mika
2020-12-30 22:30:52 -05:00
committed by GitHub
parent 6a54897577
commit 8726521604
14 changed files with 108 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rllib/models/torch/modules/convtranspose2d_stack.py
+2 -2
View File
@@ -1,8 +1,8 @@
from typing import Tuple
from ray.rllib.models.torch.misc import Reshape
from ray.rllib.models.utils import get_initializer
from ray.rllib.utils.framework import get_activation_fn, try_import_torch
from ray.rllib.models.utils import get_activation_fn, get_initializer
from ray.rllib.utils.framework import try_import_torch
torch, nn = try_import_torch()
if torch: