mirror of https://github.com/commaai/tinygrad.git
new desire shape
This commit is contained in:
parent
870ea766ee
commit
2372b262ee
|
@ -32,7 +32,7 @@ def get_random_input_tensors():
|
|||
np_inputs = {
|
||||
"input_imgs": np.random.randn(*(1, 12, 128, 256))*256,
|
||||
"big_input_imgs": np.random.randn(*(1, 12, 128, 256))*256,
|
||||
"desire": np.zeros((1,100, 8)),
|
||||
"desire": np.zeros((1,100,4)),
|
||||
"traffic_convention": np.array([[1., 0.]]),
|
||||
"features_buffer": np.random.randn(*(1, 99, 128))
|
||||
#"initial_state": np.zeros((1, 768))
|
||||
|
|
Loading…
Reference in New Issue