model(dataset['train_input'])

[Papillon-forest]

我想知道下列两段代码的区别在哪，或许有人可以点拨我一下吗？

model = model.prune()
model(dataset['train_input'])
model.plot()

model = model.prune()
model.plot()

[KindXiaoming]

上面那个可以跑通，下面那个会报错。主要是Plot之前需要forward一次获取activation。

[Papillon-forest]

@KindXiaoming Thank you for your guidance in your busy schedule, it is much appreciated!