@ -32,7 +32,7 @@ else:
TARGET_FWD_SIZE = MAX_FWD_SIZE = 384
TARGET_BWD_SIZE = 128
MAX_BWD_SIZE = 384
MAX_BWD_SIZE = 320
MAX_FWD_FEAT_SIZE = 448