Modeld: Add comment (#26188)

Rename and add comment
old-commit-hash: 0edbbdeaf9e5c8ad23b73089d2871f88ff29509e
This commit is contained in:
HaraldSchafer
2022-10-21 11:20:52 -07:00
committed by GitHub
parent 2aaefc5744
commit e7d3c730b6
+3 -1
View File
@@ -33,6 +33,8 @@ constexpr int LEAD_MHP_N = 2;
constexpr int LEAD_TRAJ_LEN = 6;
constexpr int LEAD_PRED_DIM = 4;
constexpr int LEAD_MHP_SELECTION = 3;
// Padding to get output shape as multiple of 4
constexpr int PAD_SIZE = 2;
struct ModelOutputXYZ {
float x;
@@ -232,7 +234,7 @@ constexpr int OUTPUT_SIZE = sizeof(ModelOutput) / sizeof(float);
#else
constexpr int TEMPORAL_SIZE = 0;
#endif
constexpr int NET_OUTPUT_SIZE = OUTPUT_SIZE + FEATURE_LEN + 2;
constexpr int NET_OUTPUT_SIZE = OUTPUT_SIZE + FEATURE_LEN + PAD_SIZE;
// TODO: convert remaining arrays to std::array and update model runners
struct ModelState {