Skip to content

Normalize demucs/maml behavior when using get_module#190

Merged
jansel merged 1 commit into
pytorch:masterfrom
jansel:maml202101
Jan 14, 2021
Merged

Normalize demucs/maml behavior when using get_module#190
jansel merged 1 commit into
pytorch:masterfrom
jansel:maml202101

Conversation

@jansel
Copy link
Copy Markdown
Contributor

@jansel jansel commented Jan 13, 2021

I am doing some tests where I iterate over benchmarks and do some automated changes and output correctness checking. This is a minor change to normalize the behavior of these two benchmarks.

First I make them return a Tensor() (to make checking outputs easier). Before they returned None/np.ndarray.

Second I make the following do inference (for maml it actually did training):

module, example_inputs = benchmark.get_module()
module.eval()(*example_inputs)

@jansel jansel merged commit 6f79061 into pytorch:master Jan 14, 2021
@jansel jansel deleted the maml202101 branch January 14, 2021 16:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants