search

pytorch / ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
https://pytorch-ignite.ai
BSD 3-Clause "New" or "Revised" License
4.53k stars 617 forks source link

Ignite vs Tnt vs Torchbearer #434

Closed jurukode closed 5 years ago

jurukode commented 5 years ago

Hi all,

Currently, i'm looking for high level library that can help me to simplify my Pytorch's training code. After some searches, i found these libraries are promising to try: Ignite, TNT, and Torchbearer.

After skimming the contents, i found Ignite and TNT are closely similar each other in terms of usage and the design (based on driven API design) while torchbearer offer different design (such callback etc).

Wondering if anyone have insights about those libraries? The better question would be when to use and not to use Ignite vs TNT vs Torchbearer? Any pros and cons?

Thanks!

vfdev-5 commented 5 years ago

@dyngts I think the best answer you can get is by testing all these libs by yourself :)

For example, you can checkout our examples, take one and play with it. Some other points to take into account:

PS: IMO such question should be asked at the pytorch forum. PPS: I saw you've already seen polyaxon. Here is another example to play with polyaxon and ignite.

jurukode commented 5 years ago

@vfdev-5 I wish someone already did this, hence i don't need go through all the libraries by myself 😛

I just know that you're the creator of ignite, good to know that.

PS: IMO such question should be asked at the pytorch forum.

great idea, i'll do that. Thanks!

vfdev-5 commented 5 years ago

I wish someone already did this, hence i don't need go through all the libraries by myself

@dyngts that way you'll get your own experience. All libs mostly do a similar stuff, that it is a matter of taste and community behind the project. Very soon we'll merge some notebook for playing in a more interactive way with ignite, so stay tuned :)

I just know that you're the creator of ignite, good to know that.

The creator of ignite is @alykhantejani, I'm just a frequent contributor :)

And if you have other concrete questions on the usage or you find unclear our documentation, please do not hesitate to open an issue.

williamFalcon commented 5 years ago

@dyngts Check out PyTorch Lightning as well. Comes with a lot of advanced fewtures out of the box while still keeping the research code in pure PyTorch.

Here’s a useful comparison between the three frameworks (didn’t see torchbearer when the article was written).

https://towardsdatascience.com/pytorch-lightning-vs-pytorch-ignite-vs-fast-ai-61dc7480ad8a