Skip to content

Commit 272e30b

Browse files
committed
Update the benchmark to use ftp to download from the data repository.
1 parent 5935dee commit 272e30b

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

P2B1/__main__.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import sys,os
66
import glob
77
import optparse
8+
from keras.utils.data_utils import get_file
89
HOME=os.environ['HOME']
910
def parse_list(option, opt, value, parser):
1011
setattr(parser.values, option.dest, value.split(','))
@@ -31,7 +32,7 @@ def parse_list(option, opt, value, parser):
3132
sys.exit(0)
3233
sys.path.append('home_dir')
3334

34-
import candle.candle_helper_functions as hf
35+
import candle_helper_functions as hf
3536
reload(hf)
3637
maps=hf.autoencoder_preprocess()
3738
#from keras.Helpermodules.mnist_autoencoder_helper_functions import mnist_conv_deconv_simple, mnist_conv_deconv_complex,mnist_autoencoder_preprocess,generate_figure
@@ -47,7 +48,7 @@ def parse_list(option, opt, value, parser):
4748
batch_size = 16
4849
##### Read Data ########
4950
print ('Reading Data...')
50-
data_file='%s/Work/DataSets/CANDLE/sim-numpy.npy'%HOME ### can code to read at the terminal
51+
data_file = get_file('p2_small_baseline.npy', origin='http://ftp.mcs.anl.gov/pub/candle/public/benchmarks/P2B1/p2_small_baseline.npy')
5152
print 'Data File: %s' %data_file
5253
print 'Data Format: [Num Samples, Num Molecules, Num Atoms, Position]'
5354

@@ -121,4 +122,4 @@ def step_decay(epoch):
121122
pickle.dump(loss_data,o)
122123
o.close()
123124

124-
125+

0 commit comments

Comments
 (0)