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)
示例#2
0
 def out_features(self):
     return spaces.get_max(self._out_features)
示例#3
0
 def in_features(self):
     return spaces.get_max(self._in_features)
示例#4
0
 def hidden_multiplier(self):
     return spaces.get_max(self._hidden_multiplier)
 def d_model(self):
     return spaces.get_max(self._d_model)
示例#6
0
 def in_dim(self):
     return spaces.get_max(self._in_dim)
示例#7
0
 def embed_dim(self):
     return spaces.get_max(self._embed_dim)
 def proj_dim(self):
     return spaces.get_max(self._proj_dim)
 def num_heads(self):
     return spaces.get_max(self._num_heads)