haseebs / rst-owe Goto Github PK

    for epoch in range(start_epoch, epochs + 1):
        train_batch_losses = []
        decay_lr(current_epoch=epoch, optimizer=self.optimizer)

        for entities, names, descriptions, relation_ids in tqdm(iterator(batch_size,
                                                         yield_heads=True,
                                                         yield_tails=Config.get("UseTailsToOptimize"),
                                                         cluster_relations=Config.get("ClusterRelations")),
                                                total=num_batches,
                                                desc="Train: "):