Skip to content

Commit 340944d

Browse files
committed
Update the Pilot 2 Benchmark 2 keras code to pull data from the ftp repository.
1 parent 272e30b commit 340944d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

P2B2/__main__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,11 +54,13 @@ def parse_list(option, opt, value, parser):
5454
from keras import callbacks
5555
from keras.layers.advanced_activations import ELU
5656
from keras.preprocessing.image import ImageDataGenerator
57+
from keras.utils.data_utils import get_file
5758

5859
batch_size = opts.batch_size
5960
##### Read Data ########
6061
print ('Reading Data...')
61-
data_file='%s/Research/DeepLearning/ECP CANDLE/Benchmarks/Benchmarks.git/P2B2/sim-numpy.npy'%HOME ### can code to read at the terminal
62+
data_file = get_file('p2_small_baseline.npy', origin='http://ftp.mcs.anl.gov/pub/candle/public/benchmarks/P2B2/p2_small_baseline.npy')
63+
# data_file='%s/Research/DeepLearning/ECP CANDLE/Benchmarks/Benchmarks.git/P2B2/sim-numpy.npy'%HOME ### can code to read at the terminal
6264
print 'Data File: %s' %data_file
6365
print 'Data Format: [Num Samples, Num Molecules, Num Atoms, Position]'
6466

0 commit comments

Comments
 (0)