tensorflow版cycleganでエラーが出て困っています

Question

前提・実現したいこと

pythonで、tensorflow版Cycleganを用いて声質変換を行うAIを作っているのですが、以下のようなエラーが出て困っています。

発生している問題・エラーメッセージ

2019-10-17 18:12:10.651625: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_100.dll
Traceback (most recent call last):
  File "main.py", line 4, in <module>
    tf.set_random_seed(19)
AttributeError: module 'tensorflow' has no attribute 'set_random_seed'

該当のソースコード

import argparse
import os
import tensorflow as tf
tf.set_random_seed(19)
from model import cyclegan

parser = argparse.ArgumentParser(description='')
parser.add_argument('--dataset_dir', dest='dataset_dir', default='horse2zebra', help='path of the dataset')
parser.add_argument('--epoch', dest='epoch', type=int, default=200, help='# of epoch')
parser.add_argument('--epoch_step', dest='epoch_step', type=int, default=100, help='# of epoch to decay lr')
parser.add_argument('--batch_size', dest='batch_size', type=int, default=1, help='# images in batch')
parser.add_argument('--train_size', dest='train_size', type=int, default=1e8, help='# images used to train')
parser.add_argument('--load_size', dest='load_size', type=int, default=286, help='scale images to this size')
parser.add_argument('--fine_size', dest='fine_size', type=int, default=256, help='then crop to this size')
parser.add_argument('--ngf', dest='ngf', type=int, default=64, help='# of gen filters in first conv layer')
parser.add_argument('--ndf', dest='ndf', type=int, default=64, help='# of discri filters in first conv layer')
parser.add_argument('--input_nc', dest='input_nc', type=int, default=3, help='# of input image channels')
parser.add_argument('--output_nc', dest='output_nc', type=int, default=3, help='# of output image channels')
parser.add_argument('--lr', dest='lr', type=float, default=0.0002, help='initial learning rate for adam')
parser.add_argument('--beta1', dest='beta1', type=float, default=0.5, help='momentum term of adam')
parser.add_argument('--which_direction', dest='which_direction', default='AtoB', help='AtoB or BtoA')
parser.add_argument('--phase', dest='phase', default='train', help='train, test')
parser.add_argument('--save_freq', dest='save_freq', type=int, default=1000, help='save a model every save_freq iterations')
parser.add_argument('--print_freq', dest='print_freq', type=int, default=100, help='print the debug information every print_freq iterations')
parser.add_argument('--continue_train', dest='continue_train', type=bool, default=False, help='if continue training, load the latest model: 1: true, 0: false')
parser.add_argument('--checkpoint_dir', dest='checkpoint_dir', default='./checkpoint', help='models are saved here')
parser.add_argument('--sample_dir', dest='sample_dir', default='./sample', help='sample are saved here')
parser.add_argument('--test_dir', dest='test_dir', default='./test', help='test sample are saved here')
parser.add_argument('--L1_lambda', dest='L1_lambda', type=float, default=10.0, help='weight on L1 term in objective')
parser.add_argument('--use_resnet', dest='use_resnet', type=bool, default=True, help='generation network using reidule block')
parser.add_argument('--use_lsgan', dest='use_lsgan', type=bool, default=True, help='gan loss defined in lsgan')
parser.add_argument('--max_size', dest='max_size', type=int, default=50, help='max size of image pool, 0 means do not use image pool')

args = parser.parse_args()


def main(_):
    if not os.path.exists(args.checkpoint_dir):
        os.makedirs(args.checkpoint_dir)
    if not os.path.exists(args.sample_dir):
        os.makedirs(args.sample_dir)
    if not os.path.exists(args.test_dir):
        os.makedirs(args.test_dir)

    tfconfig = tf.ConfigProto(allow_soft_placement=True)
    tfconfig.gpu_options.allow_growth = True
    with tf.Session(config=tfconfig) as sess:
        model = cyclegan(sess, args)
        model.train(args) if args.phase == 'train' \
            else model.test(args)

if __name__ == '__main__':
    tf.app.run()

試したこと

恐らく、set_random_seedという関数が見つからないということだと思うのですが、具体的にどこをどう直せば良いのかが分かりません。

補足情報

tensorflowのバージョンは1.14.0です

Comments

[ryuii]

Python3.7・tensorflow 1.40.1で、以下のコードを実行したところ正常に動作しました。
import tensorflow as tf
tf.set_random_seed(19)

tensorflow2以降では、tf.set_random_seedは名前が変わっているようです。print(tf.__version__)とPythonスクリプトを起動して、バージョンは1.40.1になっておりますでしょうか。

[RyotaYamada]

申し訳ありません。確認してみたところ、tensorflowのバージョンは2.0.0でした。

[ryuii]

回答に返信を書いておきました。

Answer 1

tensorflowのバージョンを確認してみてください。

tf.set_random_seedはtensorflow2.0以降は、
tf.random.set_seed(seed)に変わりました。

以下のコードでtensorflowのバージョンを確認してみてください。

import tensorflow as tf
print(tf.__version__)

pip installで1.14.0をインストールするときは、以下のコマンドを入力します。numpyなど依存パッケージがあるので、バージョンを合わせてください。

pip install tensorflow==1.14.0

Comments

[RyotaYamada]

お忙しい中、本当にありがとうございました！