Internal Update

PiperOrigin-RevId: 499303594
google · Jan 3, 2023 · 2b8f061 · 2b8f061
1 parent 38e2ac7
commit 2b8f061
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/deepconsensus/models/model_configs.py b/deepconsensus/models/model_configs.py
@@ -31,7 +31,6 @@
 
 from typing import Optional
 import ml_collections
-from deepconsensus.utils import dc_constants
 
 # Do not add any additional imports to the config.
 # It can lead to circular dependencies easily and should not be necessary
@@ -258,14 +257,17 @@ def get_config(config_name: Optional[str] = None) -> ml_collections.ConfigDict:
   """
   params = ml_collections.ConfigDict()
 
+  # Used for generating replicates.
+  params.trial = 1
+
   # Base config
   params.PW_MAX = 255
   params.IP_MAX = 255
   params.SN_MAX = 500
   params.STRAND_MAX = 2
 
   # Specify common configs here.
-  params.vocab_size = len(dc_constants.VOCAB)
+  params.vocab_size = 5
   params.tensorboard_update_freq = 'batch'
   params.model_checkpoint_freq = 'epoch'
   params.seed = 1