@@ -223,7 +223,7 @@ def from_config(cls, config):
223223
224224# On larger hardware, this could probably be increased considerably and
225225# Probably would improve performance ...
226- EMBEDDING_DIM = 15 # Define EMBEDDING_DIM here, to match your embedding layer.
226+ EMBEDDING_DIM = 23 # Define EMBEDDING_DIM here, to match your embedding layer.
227227
228228embedded = tf .keras .layers .Embedding (
229229 input_dim = VOCABULARY_SIZE ,
@@ -241,7 +241,7 @@ def from_config(cls, config):
241241# Just an FYI for anyone trying to apply conventional wisdom
242242# to save you the time ...
243243x = x = tf .keras .layers .Concatenate ()([embedded , position_embedding ])
244- x = tf .keras .layers .Dropout (0.6 )(x ) # AI suggested 0.4
244+ x = tf .keras .layers .Dropout (0.4 )(x ) # AI suggested 0.4
245245flattened = tf .keras .layers .Flatten ()(x )
246246
247247cerebros_base_model = tf .keras .Model (
@@ -255,23 +255,23 @@ def from_config(cls, config):
255255#
256256# Cerebros configurables
257257#
258- activation = 'gelu'
259- predecessor_level_connection_affinity_factor_first = 49.9999
260- predecessor_level_connection_affinity_factor_main = 0.31456
261- max_consecutive_lateral_connections = 22
262- p_lateral_connection = 0.39256
263- num_lateral_connection_tries_per_unit = 10
264- learning_rate = 0.0000511065
258+ activation = "relu"
259+ predecessor_level_connection_affinity_factor_first = 10
260+ predecessor_level_connection_affinity_factor_main = 40
261+ max_consecutive_lateral_connections = 20
262+ p_lateral_connection = 30
263+ num_lateral_connection_tries_per_unit = 25
264+ learning_rate = 3 * 10 ** - 3
265265epochs = 15 # [1, 100]
266- batch_size = 20
266+ batch_size = 17
267267minimum_levels = 2
268- maximum_levels = 3 # [3,7]
268+ maximum_levels = 2 # [3,7]
269269
270270minimum_units_per_level = 4
271- maximum_units_per_level = 8
271+ maximum_units_per_level = 7
272272
273273minimum_neurons_per_unit = 1
274- maximum_neurons_per_unit = 5 # [2,20]
274+ maximum_neurons_per_unit = 2
275275
276276moities_to_try = 5
277277tries_per_moity = 1
0 commit comments