TorchCraftAI
A bot for machine learning research on StarCraft: Brood War
Public Member Functions | Public Attributes | List of all members
common::WeightSummary Struct Reference

Collects metrics about a container's weights. More...

#include <debug.h>

Public Member Functions

 WeightSummary (torch::nn::Module &)
 
std::string toString () const
 

Public Attributes

long weights = 0
 
long zeroes = 0
 
long nans = 0
 
float norm1 = 0.0
 
float norm2 = 0.0
 

Detailed Description

Collects metrics about a container's weights.

Constructor & Destructor Documentation

common::WeightSummary::WeightSummary ( torch::nn::Module &  module)

Member Function Documentation

std::string common::WeightSummary::toString ( ) const

Member Data Documentation

long common::WeightSummary::nans = 0
float common::WeightSummary::norm1 = 0.0
float common::WeightSummary::norm2 = 0.0
long common::WeightSummary::weights = 0
long common::WeightSummary::zeroes = 0

The documentation for this struct was generated from the following files: