leoventu's picture
Update paper models, inference code and model card
33a1350 verified
Raw History Blame Contribute Delete
203 Bytes
"""Observation dimensions shared by the environment and the attention policy."""
N_ORDER_FEATS = 5
N_DECISION_FEATS = 10
N_GLOBAL_FEATS = 12
N_HEAD_EXTRA = N_DECISION_FEATS + N_GLOBAL_FEATS
D_EMBED = 32