fix tensorflow dep
This commit is contained in:
parent
52a7199d10
commit
f4acd286c8
@ -38,7 +38,7 @@ train_step = train.minimize(train.AdamOptimizer(1e-4), cross_entropy)
|
||||
|
||||
accuracy = reduce_mean(cast(indmax(y, 2) .== indmax(y′, 2), Float32))
|
||||
|
||||
run(session, initialize_all_variables())
|
||||
run(session, global_variables_initializer())
|
||||
|
||||
@progress for i in 1:1000
|
||||
batch = next_batch(loader, 50)
|
||||
|
@ -9,7 +9,7 @@ end
|
||||
|
||||
function makesession(model, inputs; session = Session(Graph()))
|
||||
params, stacks, output = tograph(model, inputs...)
|
||||
run(session, initialize_all_variables())
|
||||
run(session, global_variables_initializer())
|
||||
Model(model, session, params, stacks, inputs, output)
|
||||
end
|
||||
|
||||
|
@ -13,7 +13,7 @@ dt = tf(d)
|
||||
sess = TensorFlow.Session()
|
||||
X = placeholder(Float32)
|
||||
Y = Tensor(d, X)
|
||||
run(sess, initialize_all_variables())
|
||||
run(sess, global_variables_initializer())
|
||||
|
||||
@test run(sess, Y, Dict(X=>Float32.(xs))) ≈ d(xs)
|
||||
end
|
||||
|
Loading…
Reference in New Issue
Block a user