power_cogs.trainer package¶
Submodules¶
power_cogs.trainer.mnist_trainer module¶
-
class
power_cogs.trainer.mnist_trainer.
MNISTTrainer
(name: Optional[str] = None, pretrained_path: Optional[str] = None, visualize_output: bool = False, use_cuda: bool = False, device_id: int = 0, early_stoppage: bool = False, loss_threshold: float = -inf, batch_size: int = 32, epochs: int = 100, checkpoint_interval: int = 100, num_samples: Optional[int] = None, model_config: Dict[str, Any] = {}, dataset_config: Dict[str, Any] = {}, optimizer_config: Dict[str, Any] = {}, scheduler_config: Dict[str, Any] = {}, logging_config: Dict[str, Any] = {}, dataloader_config: Dict[str, Any] = {}, tune_config: Dict[str, Any] = {}, config: Dict[str, Any] = {})[source]¶