我正在使用DQN算法训练一个RL模型。在每次迭代时,我将模型保存如下:
agent = dqn.DQNTrainer(env=CustomEnv,config=config)
for n in range(100):
result = agent.train()
agent.save()
我想在不同的环境中使用经过训练的RL模型。我不知道如何在不同的环境中加载检查点并进行评估。
我尝试加载经过训练的模型(最后一个检查点),但它抛出了一个错误。我所做的工作如下:
agent.restore('./RL_saved/checkpoint-100.tune_metad
我在做西班牙语推文的情感分析。
在回顾了一些最近的文献之后,我看到最近有一项专门针对西班牙语文本(roberta-base-bne)的roberta-base-bne模型的培训工作。到目前为止,它似乎比目前最先进的西班牙语建模模型表现得更好。
对于不包括文本分类的各种任务,已经对RoBERTa模型进行了培训。我想拿这个,并微调它的文本分类,更具体地说,情感分析.
我已经完成了所有的预处理,并创建了dataset对象,并希望对模型进行本机训练。
码
# Training with native TensorFlow
from transformers import TFRobertaForS
因此,我在jupytor笔记本中实现了一个RNN单词生成器模型。当我尝试使用经过训练的模型来生成一些单词时:
with open(os.path.join(cfgs['save_dir'], 'config.pkl'), 'rb') as f:
saved_args = cPickle.load(f)
with open(os.path.join(cfgs['save_dir'], 'words_vocab.pkl'), 'rb') as f:
word
我用TensorFlow热切的模式训练了一个CNN模型。现在,我正在尝试从检查点文件中恢复经过训练的模型,但没有取得任何成功。
我发现的所有示例(如下面所示)都是关于将检查点还原到会话的。但是,我需要的是将模型恢复到急切的模式,即不创建会话。
with tf.Session() as sess:
# Restore variables from disk.
saver.restore(sess, "/tmp/model.ckpt")
基本上我需要的是:
tfe.enable_eager_execution()
model = tfe.restore('model
目前,我试图为我的公司培养一个word2vec模型。为此,我使用了上的代码,特别是。
我下载了德文维基百科转储,并从中提取文本信息。任务是用这些数据来训练一个模型。
我工作的虚拟机有Ubuntu16.04和访问特斯拉M60。周末,我训练了模型,并将检查点保存在一个单独的文件夹中。周末结束时,该模型能够回答36%的给他的评价问题(德语问题类似于“问题-词. the”)。经过培训,我想加载模型,并再次运行评估任务。为此,我在以下行中更改了代码(除了路径更改外):
with tf.Graph().as_default(), tf.Session() as session:
sa
在TensorFlow 2.2中保存模型权重后,无法加载它们。权重似乎被正确地保存了(我认为),但是,我无法加载预先训练的模型。 我当前的代码是: segmentor = sequential_model_1()
discriminator = sequential_model_2()
def save_model(ckp_dir):
# create directory, if it does not exist:
utils.safe_mkdir(ckp_dir)
# save weights
segmentor.save_weights(os.pat