from DeepJetCore.training.gpuTools import DJCSetGPUs DJCSetGPUs() import tensorflow as tf import numpy as np import time from compare_knn_outputs_op import CompareKnnOutputs from select_knn_op import SelectKnn from rknn_op import * import matplotlib matplotlib.use('Agg') import matplotlib.pyplot as plt tf.debugging.set_log_device_placement(True) def createData(nvert, ncoords): coords = tf.constant(np.random.rand(nvert, ncoords), dtype='float32') row_splits = tf.constant([0, nvert], dtype='int32') return coords, row_splits def euclidean_squared(A, B): A = tf.expand_dims(A, axis=1) #V x 1 x C B = tf.expand_dims(B, axis=0) #1 x V x C return tf.reduce_sum((A - B)**2, axis=-1) def selectNeighbours_TF(K, coords, row_splits, return_distances=False):
for s in predsamples.samples: inputdatafiles.append(s) elif args.inputSourceFileList[-6:] == ".djctd": inputdir = os.path.abspath(os.path.dirname(args.inputSourceFileList)) infile = os.path.basename(args.inputSourceFileList) inputdatafiles.append(infile) else: print('reading from text file',args.inputSourceFileList) inputdir = os.path.abspath(os.path.dirname(args.inputSourceFileList)) with open(args.inputSourceFileList, "r") as f: for s in f: inputdatafiles.append(s.replace('\n', '').replace(" ","")) DJCSetGPUs(args.gpu) custom_objs = get_custom_objects() model=load_model(args.inputModel, custom_objects=custom_objs) dc = None if args.inputSourceFileList[-6:] == ".djcdc" and not args.trainingDataCollection[-6:] == ".djcdc": dc = DataCollection(args.inputSourceFileList) if batchsize < 1: batchsize = 1 print('No training data collection given. Using batch size of',batchsize) else: dc = DataCollection(args.trainingDataCollection) outputs = [] os.system('mkdir -p '+args.outputDir)
# gpus_to_use = "1" # "comma-separated string" # machine = "V100" # gpus_to_use = "2" # "comma-separated string" allow_growing_memory = False memory_limit = -666 # Mb # ********** CONFIG end ********** # construct name of out txt-file out_textfile = "./" out_textfile += "test_flat_rndm_1Sep2021_Slicing.csv" from DeepJetCore.training.gpuTools import DJCSetGPUs if machine == "1080": DJCSetGPUs() else: DJCSetGPUs(gpus_to_use) import tensorflow as tf import numpy as np import os import time from compare_knn_outputs_op import CompareKnnOutputs from select_knn_op import SelectKnn from slicing_knn_op import SlicingKnn import matplotlib matplotlib.use('Agg') import matplotlib.pyplot as plt