hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix the tf assign error of copy the trained variable from black to white

Browse Source
This commit is contained in:
Dong Yan 2018-01-09 21:16:35 +08:00
parent c2775df8e6
commit f425085e0a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
AlphaGo/model.py
View File

@ -122,7 +122,7 @@ class ResNet(object):
print('Successfully loaded') print('Successfully loaded')
else: else:
raise ValueError("No model in path {}".format(white_checkpoint_path)) raise ValueError("No model in path {}".format(white_checkpoint_path))
self.update = [tf.assign(black_params, white_params) for black_params, white_params in self.update = [tf.assign(white_params, black_params) for black_params, white_params in
zip(self.black_var_list, self.white_var_list)] zip(self.black_var_list, self.white_var_list)]
# training hyper-parameters: # training hyper-parameters: