Searched refs:RLBuffer (Results 1 – 2 of 2) sorted by relevance
/third_party/mindspore/tests/st/ops/cpu/ |
D | test_rl_buffer_net.py | 36 class RLBuffer(nn.Cell): class 38 super(RLBuffer, self).__init__() 75 buffer = RLBuffer(batch_size=32, capcity=100, shapes=[(4,), (2,), (1,), (4,)], types=[
|
/third_party/mindspore/tests/st/ops/gpu/ |
D | test_rl_buffer_net.py | 35 class RLBuffer(nn.Cell): class 37 super(RLBuffer, self).__init__() 75 buffer = RLBuffer(batch_size=32, capcity=100, shapes=[(4,), (2,), (1,), (4,)], types=[
|