Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-05-09 21:07:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 364357405
This commit is contained in:
Rebecca Chen
2021-03-22 17:52:39 +00:00
committed by Louise Deason
parent 53326fec2e
commit d9fb969489
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
glassy_dynamics/train.py
+1 -1
View File
@@ -180,7 +180,7 @@ def train_model(train_file_pattern: Text,
mlp_kwargs: Optional[Dict[Text, Any]] = None, mlp_kwargs: Optional[Dict[Text, Any]] = None,
edge_threshold: float = 2.0, edge_threshold: float = 2.0,
measurement_store_interval: int = 1000, measurement_store_interval: int = 1000,
checkpoint_path: Optional[Text] = None) -> float: checkpoint_path: Optional[Text] = None) -> float: # pytype: disable=annotation-type-mismatch
"""Trains GraphModel using tensorflow. """Trains GraphModel using tensorflow.
Args: Args: