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)