Add ForwardContext
to wrap model forward pass
#267
Merged
ForwardContext
to wrap model forward pass
#267