-
Notifications
You must be signed in to change notification settings - Fork 445
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[gala kinematic] split train and test stages #1751
base: gala_kinematic
Are you sure you want to change the base?
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I left some comments.
@@ -369,7 +369,7 @@ EpisodeSet generateBenchmarkEpisodeSet(const EpisodeGeneratorConfig& config, | |||
|
|||
// distribute stages across episodes | |||
for (int i = 0; i < numEpisodes; i++) { | |||
int stageIndex = i * config.numStageVariations / numEpisodes; | |||
int stageIndex = i * (config.maxStageNumber - config.minStageNumber) / numEpisodes; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
off-by-one here?
int numStageVariations = 12; // see selectedReplicaCadBakedStages | ||
// int numStageVariations = 12; // see selectedReplicaCadBakedStages | ||
int minStageNumber = 0; | ||
int maxStageNumber = 12; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
off-by-one here? previously, the max stage index was 11, because there are only 12 stages.
@@ -13,7 +13,9 @@ namespace batched_sim { | |||
struct EpisodeGeneratorConfig { | |||
int numEpisodes = 100; | |||
int seed = 3; // this is 3 for legacy reason | |||
int numStageVariations = 12; // see selectedReplicaCadBakedStages | |||
// int numStageVariations = 12; // see selectedReplicaCadBakedStages | |||
int minStageNumber = 0; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
"index" is probably a better term than "number"
.def_readwrite("num_stage_variations", &EpisodeGeneratorConfig::numStageVariations, R"(Todo)") | ||
// .def_readwrite("num_stage_variations", &EpisodeGeneratorConfig::numStageVariations, R"(Todo)") | ||
.def_readwrite("min_stage_number", &EpisodeGeneratorConfig::minStageNumber, R"(Todo)") | ||
.def_readwrite("max_stage_number", &EpisodeGeneratorConfig::maxStageNumber, R"(Todo)") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
FYI, the docstring is optional...
Motivation and Context
How Has This Been Tested
Types of changes
Checklist