7d1e2acfea6134d18dd5f3db366b23e2d3f6e3a8,tests/performance_test/vgg/tf2-eager.py,,,#,8

Before Change

import tensorflow as tf
from exp_config import random_input_generator, MONITOR_INTERVAL, NUM_ITERS, BATCH_SIZE, LERANING_RATE

tf.config.gpu.set_per_process_memory_growth(True)

// get the whole model
vgg = VGG16(weights=None)

After Change

import tensorflow as tf
from exp_config import random_input_generator, MONITOR_INTERVAL, NUM_ITERS, BATCH_SIZE, LERANING_RATE

gpus = tf.config.experimental.list_physical_devices("GPU")
if gpus:
    for gpu in gpus:
        tf.config.experimental.set_memory_growth(gpu, True)

// get the whole model
vgg = VGG16(weights=None)

// system monitor
info = psutil.virtual_memory()

Instances