def validation_epoch_end(self, outputs):
        val_loss = mean(outputs, 'val_loss')
        val_acc1 = mean(outputs, 'val_acc1')
        val_acc5 = mean(outputs, 'val_acc5')

        log = {'val_loss': val_loss, 'val_acc1': val_acc1, 'val_acc5': val_acc5}
        self.log_dict(log)
示例#2
0
    def validation_epoch_end(self, outputs):
        val_loss = mean(outputs, "val_loss")
        val_acc1 = mean(outputs, "val_acc1")
        val_acc5 = mean(outputs, "val_acc5")

        log = {"val_loss": val_loss, "val_acc1": val_acc1, "val_acc5": val_acc5}
        self.log_dict(log)
示例#3
0
    def validation_epoch_end(self, outputs):
        val_loss = mean(outputs, 'val_loss')
        val_acc1 = mean(outputs, 'val_acc1')
        val_acc5 = mean(outputs, 'val_acc5')

        self.log('val_loss_agg', val_loss)
        self.log('val_acc1_agg', val_acc1)
        self.log('val_acc5_agg', val_acc5)
示例#4
0
    def test_epoch_end(self, outputs):
        test_loss = mean(outputs, 'test_loss')
        test_acc1 = mean(outputs, 'test_acc1')
        test_acc5 = mean(outputs, 'test_acc5')

        return {
            'test_lincls_loss': test_loss.item(),
            'test_lincls_acc1': test_acc1.item(),
            'test_lincls_acc5': test_acc5.item(),
        }
    def validation_epoch_end(self, outputs):
        val_loss = mean(outputs, 'val_loss')
        val_acc1 = mean(outputs, 'val_acc1')
        val_acc5 = mean(outputs, 'val_acc5')

        log = {
            'val_loss': val_loss,
            'val_acc1': val_acc1,
            'val_acc5': val_acc5
        }
        return {'val_loss': val_loss, 'log': log, 'progress_bar': log}
示例#6
0
    def validation_epoch_end(self, outputs):
        val_nce = metrics.mean(outputs, 'val_nce')

        log = {'val_nce_loss': val_nce}
        if self.online_evaluator:
            mlp_acc = metrics.mean(outputs, 'mlp_acc')
            mlp_loss = metrics.mean(outputs, 'mlp_loss')
            log['val_mlp_acc'] = mlp_acc
            log['val_mlp_loss'] = mlp_loss

        return {'val_loss': val_nce, 'log': log}
示例#7
0
    def validation_epoch_end(self, outputs):
        val_loss = mean(outputs, "val_loss")
        val_acc1 = mean(outputs, "val_acc1")
        val_acc5 = mean(outputs, "val_acc5")

        log = {
            "val_loss": val_loss,
            "val_acc1": val_acc1,
            "val_acc5": val_acc5,
        }
        return {"val_loss": val_loss, "log": log, "progress_bar": log}
示例#8
0
    def test_epoch_end(self, outputs):
        test_loss = mean(outputs, 'test_loss')
        test_acc1 = mean(outputs, 'test_acc1')
        test_acc5 = mean(outputs, 'test_acc5')

        results = {
            'test_con_loss': test_loss.item(),
            'test_con_acc1': test_acc1.item(),
            'test_con_acc5': test_acc5.item(),
        }

        return results
示例#9
0
    def validation_epoch_end(self, outputs: list):
        val_loss = mean(outputs, 'val_loss')

        log = dict(val_loss=val_loss, )

        progress_bar = {}
        if self.online_evaluator:
            mlp_acc = mean(outputs, 'mlp_acc')
            mlp_loss = mean(outputs, 'mlp_loss')
            log['val_mlp_acc'] = mlp_acc
            log['val_mlp_loss'] = mlp_loss
            progress_bar['val_acc'] = mlp_acc

        return dict(val_loss=val_loss, log=log, progress_bar=progress_bar)
示例#10
0
    def validation_epoch_end(self, results):
        val_loss = mean(results, "val_loss")
        val_acc1 = mean(results, "val_acc1")
        val_acc5 = mean(results, "val_acc5")

        log = {"val_loss": val_loss, "val_acc1": val_acc1, "val_acc5": val_acc5}
        self.log_dict(log, sync_dist=True)
        try:
            # LightningModule.print() has an issue on printing when validation mode
            self.print(
                f"[Epoch {self.current_epoch}]: [Val loss: {val_loss} / Val acc1: {val_acc1} / Val acc5: {val_acc5}]"
            )
        except:
            pass