flambe.experiment.tune_adapter

Module Contents

class flambe.experiment.tune_adapter.TuneAdapter[source]

Bases: ray.tune.Trainable

Adapter to the tune.Trainable inteface.

_setup(self, config: Dict)[source]

Subclasses should override this for custom initialization.

save(self, checkpoint_dir: Optional[str] = None)[source]

Override to replace checkpoint.

_train(self)[source]

Subclasses should override this to implement train().

_save(self, checkpoint_dir: str)[source]

Subclasses should override this to implement save().

_restore(self, checkpoint: str)[source]

Subclasses should override this to implement restore().

_stop(self)[source]

Subclasses should override this for any cleanup on stop.