Added color!

Highlight the non-containers with '*' in profiler.print() to ease the reading.

Update.

So, back to decorating the classes and not the objects so that torch.save() does not complain with SpatialConvolution. Added the possibility to pass the total time to profiler.print() so that the fraction of time used by the different functions can be displayed.

Fixed a stupid and embarrassing typo.

Update.

Tidied a bit the printing.

Update.

Decorating the model themselves (and not the class) makes more sense as it allows to decorate only some models or parts of models, which may be desirable sometimes.

Cosmetics.

Update.

This file should not be executable.

Initial commit.

Update.

Now decorate function as high as possible in the class hierarchy, and breaks down timing by function.

Changed the timing display.

Added comments, a slightly more complicated model, and the header.

Cosmetics + added a header.

Update.

Initial commit