hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-05 10:01:43 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Correctly setting the input to Train (#3853)

Browse source 
In the ResNetTrainActor class, the data are now exactly build using the Train flag for the cifar_input script.
This commit is contained in:
LorenzoCevolani 2019-07-27 20:08:35 +02:00 • committed by Richard Liaw
parent a62c5f40f6
commit 10cbcced7e
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
examples/resnet/resnet_main.py
View file

@ -107,7 +107,7 @@ class ResNetTrainActor(object):
with tf.device("/gpu:0" if num_gpus > 0 else "/cpu:0"):
# Build the model.
images, labels = cifar_input.build_input(data, hps.batch_size,
dataset, False)
dataset, True)
self.model = resnet_model.ResNet(hps, images, labels, "train")
self.model.build_graph()
config = tf.ConfigProto(allow_soft_placement=True)