Too many values to unpack

[SanphouWang]

code at src/server line 259: in_max_idx, lodo_max_idx, _ = np.argmax(key_metric, axis=0) raises a ValueError: too many values to unpack.
I think there is a error in the code of this line becuase np.argmax(key_metric, axis=0) returns a tensor with shape (1,4), and this line should be corrected as in_max_idx, lodo_max_idx, _ , _= np.argmax(key_metric, axis=0)