Skip to content

T
tic
Commit 7f942474 by Nick Hynes Committed by Tianqi Chen
Options

[Relay] Fix name of bias in testing.mlp (#2892)

parent 23aa24cf
Showing with 1 additions and 1 deletions
python/tvm/relay/testing/mlp.py
......@@ -50,7 +50,7 @@ def get_net(batch_size,
dtype=dtype)
data = relay.nn.batch_flatten(data)
fc1 = relay.nn.dense(data, relay.var("fc1_weight"), units=128)
fc1 = relay.nn.bias_add(fc1, relay.var("fc2_bias"))
fc1 = relay.nn.bias_add(fc1, relay.var("fc1_bias"))
act1 = relay.nn.relu(fc1)
fc2 = relay.nn.dense(act1, relay.var("fc2_weight"), units=64)
fc2 = relay.nn.bias_add(fc2, relay.var("fc2_bias"))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment