![]() add_state ( name, default, dist_reduce_fx = None, persistent = False ) Īdds metric state variable. Initializes internal Module state, shared by both nn.Module and ScriptModule. default: None (which selects the entire world)ĭist_sync_fn ¶ ( Optional) – Callback that performs the allgather operation on the metric state. Process_group ¶ ( Optional) – Specify the process group on which synchronization is called. default: Trueĭist_sync_on_step ¶ ( bool) – Synchronize metric state across processes at each forward() ParametersĬompute_on_step ¶ ( bool) – Forward only calls update() and returns None if this is set to False. A call to forward()Īutomatically calls update() and also returns the metric value at the current step. ![]() Is valid, but it won’t return the metric value at the current step. ![]() Different metrics only override update() and not forward(). ![]()
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |