c0f75b3a70cc2e433472de9dfd2406c685b561a9,test/test_item_ranking.py,,,#,33

Before Change

    train_data, test_data, n_user, n_item = load_data_neg(test_size=0.2, sep="\t")

    config = tf.ConfigProto()
    config.gpu_options.allow_growth = True

    with tf.Session(config=config) as sess:
        model = None

After Change

    try:
        gpus = tf.config.experimental.list_physical_devices("GPU")
        tf.config.experimental.set_memory_growth(gpus[0], True)
    except:
        // Invalid device or cannot modify virtual devices once initialized.
        pass

Instances