pytorchbearer/torchbearer

Make StateKey objects functions of state

ethanwharris opened this issue · 0 comments

It would be easy to make it so that state keys can be used as losses, they should just return state[self] when called