habitat
habitat copied to clipboard
Is it possible to provide the code generated by different model variants with different hyperparameter configurations
Networks with different kernel size, stride, padding, etc. configurations and their execution time label
Thanks for your interest in Habitat!
Can you clarify what you mean exactly by "code generated by different model variants"?
It's the code that collects the dataset. For networks with different configurations (different kernel size, batch size , channel size, etc.), how their execution time label are collected
The code we used to collect the training data is under the tools/recording/
subdirectory. For example, you can see how the Conv2d
training data was collected here. You should be able to modify that code if you want to collect training data for a different operation or with different configurations.