【问题标题】:Is there anyway to run GPT2 without GPU and TensorFlow无论如何在没有 GPU 和 TensorFlow 的情况下运行 GPT2
【发布时间】:2020-01-04 06:18:00
【问题描述】:

GPT2 是一个出色的 NLP OpenAI 项目。开发人员要求声明我们需要使用张量流和 GPU。我只想使用(而不是训练)现有的训练参数。有没有什么方法可以使用 GPT2 而不需要昂贵的硬件和 GPU 并且不需要安装 Tensor Flow?

【问题讨论】:

    标签: python tensorflow nlp gpu


    【解决方案1】:

    我会做什么:

    • sess.run([var for var in tf.trainable_variables]) 获取经过训练的参数并将它们保存为 numpy 数组。
    • 重建一个仅 numpy 的 gpt2 版本。

    但是:

    • 重建模型需要大量工作
    • 您需要至少安装一次 tensorflow 才能获得经过训练的变量。

    至于gpu,你实际上并不需要它,模型将使用tensorflow的cpu版本运行(特别需要gpu来加快训练时间)。

    【讨论】:

      猜你喜欢
      • 2022-01-03
      • 1970-01-01
      • 1970-01-01
      • 2019-02-01
      • 1970-01-01
      • 1970-01-01
      • 2021-05-16
      • 2010-12-17
      • 2018-04-29
      相关资源
      最近更新 更多