掩码RCNN内存管理

Question

我正在尝试将模型拟合到内存中。 我看到一种可能的解决方案是：

import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.allow_growth = True

或这个孔穴：

#config.gpu_options.per_process_gpu_memory_fraction = 0.4 #Allocate fixed memory

最后

set_session(tf.Session(config=config))

我想知道我应该把它放在哪里？