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