@ -311,7 +311,6 @@ def _ssd_test(args: argparse.Namespace) -> None:
|
|||||||
from twomartens.masterthesis.ssd_keras.keras_layers import keras_layer_L2Normalization
|
from twomartens.masterthesis.ssd_keras.keras_layers import keras_layer_L2Normalization
|
||||||
from twomartens.masterthesis.ssd_keras.keras_loss_function import keras_ssd_loss
|
from twomartens.masterthesis.ssd_keras.keras_loss_function import keras_ssd_loss
|
||||||
|
|
||||||
|
|
||||||
config = tf.ConfigProto()
|
config = tf.ConfigProto()
|
||||||
config.log_device_placement = False
|
config.log_device_placement = False
|
||||||
config.gpu_options.allow_growth = False
|
config.gpu_options.allow_growth = False
|
||||||
|
|||||||
Reference in New Issue
Block a user