power_cogs.config.base package¶
Submodules¶
power_cogs.config.base.base_config module¶
-
class
power_cogs.config.base.base_config.
BaseDatasetConfig
(_target_: 'str' = '???')[source]¶ Bases:
object
-
class
power_cogs.config.base.base_config.
BaseLoggingConfig
(checkpoint_path: 'str' = 'checkpoints', tensorboard_log_path: 'Optional[str]' = None)[source]¶ Bases:
object
-
checkpoint_path
= 'checkpoints'¶
-
tensorboard_log_path
= None¶
-
-
class
power_cogs.config.base.base_config.
BaseModelConfig
(_target_: 'str' = '???')[source]¶ Bases:
object
-
class
power_cogs.config.base.base_config.
BaseTrainerConfig
(defaults: 'List[Any]' = <factory>, _target_: 'str' = '???', name: 'Optional[str]' = None, pretrained_path: 'Optional[str]' = None, visualize_output: 'bool' = True, 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: 'Any' = '???', dataset_config: 'Any' = '???', optimizer_config: 'Any' = '???', scheduler_config: 'Any' = '???', logging_config: 'Any' = '???', dataloader_config: 'Any' = '???', tune_config: 'Any' = '???', config: 'Any' = <factory>)[source]¶ Bases:
object
-
batch_size
= 32¶
-
checkpoint_interval
= 100¶
-
dataloader_config
= '???'¶
-
dataset_config
= '???'¶
-
device_id
= 0¶
-
early_stoppage
= False¶
-
epochs
= 100¶
-
logging_config
= '???'¶
-
loss_threshold
= -inf¶
-
model_config
= '???'¶
-
name
= None¶
-
num_samples
= None¶
-
optimizer_config
= '???'¶
-
pretrained_path
= None¶
-
scheduler_config
= '???'¶
-
tune_config
= '???'¶
-
use_cuda
= False¶
-
visualize_output
= True¶
-
Module contents¶
-
class
power_cogs.config.base.
BaseLoggingConfig
(checkpoint_path: 'str' = 'checkpoints', tensorboard_log_path: 'Optional[str]' = None)[source]¶ Bases:
object
-
checkpoint_path
= 'checkpoints'¶
-
tensorboard_log_path
= None¶
-
-
class
power_cogs.config.base.
BaseTrainerConfig
(defaults: 'List[Any]' = <factory>, _target_: 'str' = '???', name: 'Optional[str]' = None, pretrained_path: 'Optional[str]' = None, visualize_output: 'bool' = True, 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: 'Any' = '???', dataset_config: 'Any' = '???', optimizer_config: 'Any' = '???', scheduler_config: 'Any' = '???', logging_config: 'Any' = '???', dataloader_config: 'Any' = '???', tune_config: 'Any' = '???', config: 'Any' = <factory>)[source]¶ Bases:
object
-
batch_size
= 32¶
-
checkpoint_interval
= 100¶
-
dataloader_config
= '???'¶
-
dataset_config
= '???'¶
-
device_id
= 0¶
-
early_stoppage
= False¶
-
epochs
= 100¶
-
logging_config
= '???'¶
-
loss_threshold
= -inf¶
-
model_config
= '???'¶
-
name
= None¶
-
num_samples
= None¶
-
optimizer_config
= '???'¶
-
pretrained_path
= None¶
-
scheduler_config
= '???'¶
-
tune_config
= '???'¶
-
use_cuda
= False¶
-
visualize_output
= True¶
-