modules

This module is designed for recording:

  1. support Recorder. Class WithRecorder allows to use recorder to save statistics during training using csv and tensorboard.

Class

WithRecorder

class gedml.core.modules.with_recorder.WithRecorder(**kwargs)[source]

Bases: torch.nn.modules.module.Module

Work with recorder protocol.

add_recordable_attr(name: str)[source]
Parameters

name (str) – The name of attributes.