Restore subset of variables in Tensorflow

后端 未结 4 1637
名媛妹妹
名媛妹妹 2020-12-08 01:17

I am training a Generative Adversarial Network (GAN) in tensorflow, where basically we have two different networks each one with its own optimizer.

self.G, s         


        
4条回答
  •  没有蜡笔的小新
    2020-12-08 02:05

    You can create a separate instance of tf.train.Saver() with the var_list argument set to the variables you want to restore. And create a separate instance to save the variables

提交回复
热议问题