INNER CODE UNIT · Python

dev_cost

ShannonAI/glyce · glyce/models/latticeLSTM/main.py:331

        dev_cost = dev_finish - epoch_finish

        if seg:
            current_score = f
            logger.info(("Dev: time: %.2fs, speed: %.2fst/s; acc: %.4f, p: %.4f, r: %.4f, f: %.4f" % (dev_cost, speed, acc, p, r, f)))
        else:
            current_score = acc
            logger.info(("Dev: time: %.2fs speed: %.2fst/s; acc: %.4f"%(dev_cost, speed, acc)))

        if current_score > best_dev:
            if seg:
                logger.info(F"Exceed previous best f score: {best_dev}")
            else:
                logger.info(F"Exceed previous best acc score: {best_dev}")
            model_name = os.path.join(save_model_dir, 'saved.model')
            torch.save(model.state_dict(), model_name)
            best_dev = current_score 
        # ## decode test

View source record →