やりたいこと

Chainer公式サンプルのSeq2Seqモデルを実行してみました。
実行自体は問題なく完了したので、
そこでさらに、計算グラフの可視化を行おうとしました。

問題

可視化にあたり、以下のようにchainer.training.extensions.dump_graph()を組み込んでみました。

python:
1...
2from chainer.training import extensions
3...
4trainer = training.Trainer(updater, (args.epoch, 'epoch'), out=args.out)
5trainer.extend(extensions.dump_graph('main/loss'))
6...

しかし実行してみると、下記のようなエラーメッセージが現れて処理が止まります。

Exception in main training loop: root value is not a Variable
Traceback (most recent call last):
  File "/root/anaconda3/lib/python3.6/site-packages/chainer/training/trainer.py", line 302, in run
    entry.extension(self)
  File "/root/anaconda3/lib/python3.6/site-packages/chainer/training/extensions/computational_graph.py", line 82, in dump_graph
    raise TypeError('root value is not a Variable')
Will finalize trainer extensions and updater before reraising the exception.
Traceback (most recent call last):
  File "seq2seq-attn.py", line 325, in <module>
    main()
  File "seq2seq-attn.py", line 321, in main
    trainer.run()
  File "/root/anaconda3/lib/python3.6/site-packages/chainer/training/trainer.py", line 313, in run
    six.reraise(*sys.exc_info())
  File "/root/anaconda3/lib/python3.6/site-packages/six.py", line 686, in reraise
    raise value
  File "/root/anaconda3/lib/python3.6/site-packages/chainer/training/trainer.py", line 302, in run
    entry.extension(self)
  File "/root/anaconda3/lib/python3.6/site-packages/chainer/training/extensions/computational_graph.py", line 82, in dump_graph
    raise TypeError('root value is not a Variable')
TypeError: root value is not a Variable

別のサンプル（MNIST）では同様のコードで問題なくグラフが出力できています。
（サンプル中にデフォルトでdump_graph()が組み込まれています。）
どうしてseq2seq.pyの場合には失敗するのでしょうか。

無事にseq2seq.pyでも計算グラフを出力できたという方がいたら、アドバイスをいただきたいと思います。

よろしくお願いします。