Skip to content

Add track_running_stats flag to config parameters for GNN Training

Jared Dynes Burleson requested to merge track_running_stats into dev

Add an option for a flag during GNN training config to turn on 'track_running_stats' for 'batchnorm'. The default option is False, but allow config to set the value to True.

Merge request reports