def __init__(self):
Object.__init__(self)
ABC.__init__(self)
self.model_output_folder = f"outputs/weights/{self.config.run_id}"
self.result_output_folder = f"outputs/results/{self.config.run_id}"
if self.config.save_best_model and not os.path.exists(
self.model_output_folder):
os.mkdir(self.model_output_folder)
if self.config.save_results and not os.path.exists(
self.result_output_folder):
os.mkdir(self.result_output_folder)
self.model: Optional[Union[Type[Model], nn.DataParallel]] = None
self.optimizer: Optional[Type[Optimizer]] = None
self.logger.info(f"Loading data mapping...")
manifest_file = f"{self.config.data_lookup}"
self.mapping: Mapping = Mapping(mapping_path=manifest_file)
self.logger.info(f"Building label encoder...")
labels = list(map(lambda x: x.label, self.mapping))
self.label_encoder = LabelEncoder()
self.label_encoder.fit(labels)
self.logger.info(
f"Created encoder with classes: {self.label_encoder.classes_}.")
def __init__(self):
Object.__init__(self)
ABC.__init__(self)
self.dataset: Optional[Dataset] = Dataset(self.config.dataset_path)
self.model: Optional[Union[Model, DataParallel]] = self.build_model()
self.optimizer: Optional[Optimizer] = Engine.build_optimizer(
**self.build_optimizer_params())
self.lr_scheduler: Optional[
lr_scheduler._LRScheduler] = self.build_lr_scheduler()
def __init__(self, mapping: List[ImageRecord] = None, mapping_path: str = None):
Object.__init__(self)
Iterable.__init__(self)
assert mapping is not None or mapping_path is not None, \
"Must provide either a list of ImageRecord objects or a mapping path where such list can " \
"be constructed from."
self.mapping: List[ImageRecord] = mapping or self._load_mapping(mapping_path)
self.iter_idx = 0
def __init__(self, num_nodes: int, num_relations: int,
triplets: np.ndarray):
Object.__init__(self)
DGLGraph.__init__(self)
assert triplets.shape[0] == 3
src, rel, dst = triplets
self.add_nodes(num_nodes)
self.add_edges(src, dst)
self.set_n_initializer(zero_initializer)
self.set_e_initializer(zero_initializer)
self.logger.info(
f"Constructed graph with {num_nodes} nodes and {num_relations} relation types."
)
def __init__(self):
Object.__init__(self)
nn.Module.__init__(self)
ABC.__init__(self)
