Would a TimedModule decorator be useful?

[juesato]

The interface would be something like nn.TimedModule(module, print_interval, name), and after print_interval calls to the module, it would print name and the average time for a forwards and backwards pass. If this seems useful, I'm happy to implement it here, or if there are other ways of timing specific modules / technical issues I'm overlooking, that'd be useful too.

[nicholas-leonard]

That would indeed be useful!

[nicholas-leonard]

I would call it nn.Profile, but the naming is up to you.

[juesato]

Sorry for the delay on this - I wrote something, but the results look strange to me. I probably won't work on it this week, but I'll submit a PR this weekend.