Exemplo n.º 1
0
 def test_integer(self):
     space = Integer(lower=1, upper=4)
     for i in range(4):
         self.assertEqual(space[i], i + 1)
     self.assertEqual("Integer(lower=1, upper=4, default=None)", str(space))
     self.assertEqual(get_max(space), 4)
     self.assertEqual(get_min(space), 1)
Exemplo n.º 2
0
 def out_features(self):
     return spaces.get_max(self._out_features)
Exemplo n.º 3
0
 def in_features(self):
     return spaces.get_max(self._in_features)
Exemplo n.º 4
0
 def hidden_multiplier(self):
     return spaces.get_max(self._hidden_multiplier)
Exemplo n.º 5
0
 def d_model(self):
     return spaces.get_max(self._d_model)
Exemplo n.º 6
0
 def in_dim(self):
     return spaces.get_max(self._in_dim)
Exemplo n.º 7
0
 def embed_dim(self):
     return spaces.get_max(self._embed_dim)
Exemplo n.º 8
0
 def proj_dim(self):
     return spaces.get_max(self._proj_dim)
Exemplo n.º 9
0
 def num_heads(self):
     return spaces.get_max(self._num_heads)