tabensemb.model.AbstractNN.print

method

AbstractNN.print(*args: Any, **kwargs: Any) → None

Prints only from process 0. Use this in any distributed mode to log only once.

Parameters:

• *args – The thing to print. The same as for Python’s built-in print function.

• **kwargs – The same as for Python’s built-in print function.

Example:

def forward(self, x):
    self.print(x, 'in forward')