小编And*_*ino的帖子

我在以下代码中创建了一个 tf-agent DqnAgent：

tf_agent = dqn_agent.DqnAgent(
    train_env.time_step_spec(),
    train_env.action_spec(),
    q_network=q_net,
    optimizer=optimizer,
    td_errors_loss_fn=dqn_agent.element_wise_squared_loss,
    train_step_counter=train_step_counter

)

在训练循环期间，我保存了这个模型

tf.saved_model.save(tf_agent, saved_models_path)

训练后，我想加载保存的模型

if tf.saved_model.contains_saved_model(saved_models_path):
    tf_agent = tf.saved_model.load(saved_models_path)

此代码仅在文件夹中saved_path包含一个时才会加载保存的模型，函数contains_saved_model(saved_models_path)返回True，因此模型已加载，但出现异常并且程序崩溃：

Traceback (most recent call last):
    File "/home/claudino/Projetos/dino-tf-agents/dino_ia/model/agent.py", line 50, in <module>
        tf_agent = tf.saved_model.load(saved_models_path)
    File "/home/claudino/Projetos/dino-tf-agents/venv/lib/python3.6/site-packages/tensorflow/python/saved_model/load.py", line 408, in load
        return load_internal(export_dir, tags)
    File "/home/claudino/Projetos/dino-tf-agents/venv/lib/python3.6/site-packages/tensorflow/python/saved_model/load.py", line 432, in load_internal
        export_dir)
    File "/home/claudino/Projetos/dino-tf-agents/venv/lib/python3.6/site-packages/tensorflow/python/saved_model/load.py", line 58, in __init__
        self._load_all()
    File "/home/claudino/Projetos/dino-tf-agents/venv/lib/python3.6/site-packages/tensorflow/python/saved_model/load.py", line 168, in _load_all
        slot_variable = optimizer_object.add_slot(
    AttributeError: '_UserObject' object has no …

我遇到了类似的问题

但我使用的是 scala 2.11.1、slick 3.2.0，并使用 SBT 手动编译，而不是使用 IntelliJ。

我已经为数据库定义了一个 dntity：

案例类 ScheduleItem(id:Option[Int], cron:String, script:String,created_at:Long,updated_at:Long,created_by:String,updated_by:String)

object ScheduleItems {

  class ScheduleItemsT(tag: Tag) extends Table[ScheduleItem](tag, "schedule_items") {

    def id = column[Int]("id", O.PrimaryKey, O.AutoInc)

    def cron = column[String]("column")

    def script = column[String]("script")

    def created_at = column[Long]("created_at", SqlType("timestamp without time zone"))

    def updated_at = column[Long]("updated_at", SqlType("timestamp without time zone"))

    def created_by = column[String]("created_by")

    def updated_by = column[String]("updated_by")

    def * = (id.?, cron, script, created_at, updated_at, created_by, updated_by) <> (ScheduleItem.tupled, ScheduleItem.unapply)
  }

  val table = …