memorax.loggers.WandbLogger

memorax.loggers.WandbLogger#

class memorax.loggers.WandbLogger[source]#

Bases: BaseLogger[WandbLoggerState]

WandbLogger(entity: Optional[str] = None, project: Optional[str] = None, name: Optional[str] = None, group: Optional[str] = None, mode: Literal[‘online’, ‘disabled’, ‘shared’] = ‘disabled’, num_seeds: int = 1)

entity: str | None = None#
project: str | None = None#
name: str | None = None#
group: str | None = None#
mode: Literal['online', 'disabled', 'shared'] = 'disabled'#
num_seeds: int = 1#
init(**kwargs)[source]#
Return type:

WandbLoggerState

log(state, data, step)[source]#
Return type:

WandbLoggerState

Parameters:

  • state (WandbLoggerState)

  • data (Any)

  • step (int)

emit(state)[source]#
Return type:

WandbLoggerState

Parameters:

state (WandbLoggerState)

finish(state)[source]#
Return type:

None

Parameters:

state (WandbLoggerState)

classmethod __class_getitem__(params)#

Parameterizes a generic class.

At least, parameterizing a generic class is the main thing this method does. For example, for some generic class Foo, this is called when we do Foo[int] - there, with cls=Foo and params=int.

However, note that this method is also called when defining generic classes in the first place with class Foo(Generic[T]): ….

__init__(entity=None, project=None, name=None, group=None, mode='disabled', num_seeds=1)#
Parameters:

  • entity (str | None)

  • project (str | None)

  • name (str | None)

  • group (str | None)

  • mode (Literal['online', 'disabled', 'shared'])

  • num_seeds (int)

Return type:

None

from_tuple()#
get(k[, d]) → D[k] if k in D, else d.  d defaults to None.#
items() → a set-like object providing a view on D's items#
keys() → a set-like object providing a view on D's keys#
replace(**kwargs)#

Returns a new object replacing the specified fields with new values.

to_tuple()#
values() → an object providing a view on D's values#
Contents