new desire shape

This commit is contained in:
Yassine Yousfi 2022-10-17 15:20:38 -07:00
parent 870ea766ee
commit 2372b262ee
1 changed files with 1 additions and 1 deletions

View File

@ -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))