We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1bf8c66 commit 5fdcc66Copy full SHA for 5fdcc66
src/sagemaker/rl/estimator.py
@@ -42,8 +42,8 @@
42
"0.5": {"tensorflow": "1.11"},
43
"0.6.5": {"tensorflow": "1.12"},
44
"0.6": {"tensorflow": "1.12"},
45
- "0.8.2":{"tensorflow": "2.1"},
46
- "0.8.5":{"tensorflow": "2.1", "pytorch": "1.5"}
+ "0.8.2": {"tensorflow": "2.1"},
+ "0.8.5": {"tensorflow": "2.1", "pytorch": "1.5"}
47
},
48
}
49
@@ -291,7 +291,7 @@ def train_image(self):
291
self._image_version(),
292
py_version="py36",
293
account=DEFAULT_RL_ACCOUNT
294
- )
+ )
295
296
return fw_utils.create_image_uri(
297
self.sagemaker_session.boto_region_name,
0 commit comments