Skip to main content

Crate burn_central_macros

Crate burn_central_macros 

Source
Expand description

§Burn Central Macros

As define in the burn central crate documentation, this crate provide the macros to register a functions. You probably don’t need more information to it then that, but if you do we got you covered.

§Role and responsablity

The role of the macros is really light weight. We don’t want to make the register crate the hearth of our runtime. So it simply wrap your function into another functions define in the runtime and the runtime does the rest of the magic.

§Usage

To use the macros you simply need to import it from this crate and use the register macro to mark your training and inference functions. Here is an example:

#[register(training, name = "my_training_procedure")]
async fn my_training_function() {
 // Your training code here
}

Attribute Macros§

register
Macro to register your training and inference functions.