mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 23:09:51 +08:00
Update Cloudpickle (#5643)
This commit is contained in:
committed by
Philipp Moritz
parent
ed761900f6
commit
0010f54378
@@ -2,4 +2,4 @@ from __future__ import absolute_import
|
||||
|
||||
from ray.cloudpickle.cloudpickle import *
|
||||
|
||||
__version__ = '0.8.0.dev0'
|
||||
__version__ = '1.2.2.dev0'
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@@ -84,6 +84,13 @@ class RNNSpyModel(Model):
|
||||
capture_index = 0
|
||||
|
||||
def _build_layers_v2(self, input_dict, num_outputs, options):
|
||||
# Previously, a new class object was created during
|
||||
# deserialization and this `capture_index`
|
||||
# variable would be refreshed between class instantiations.
|
||||
# This behavior is no longer the case, so we manually refresh
|
||||
# the variable.
|
||||
RNNSpyModel.capture_index = 0
|
||||
|
||||
def spy(sequences, state_in, state_out, seq_lens):
|
||||
if len(sequences) == 1:
|
||||
return 0 # don't capture inference inputs
|
||||
|
||||
Reference in New Issue
Block a user