Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Convert TweetyNet to Pytorch model #30

Merged
merged 5 commits into from
Feb 9, 2020
Merged

Convert TweetyNet to Pytorch model #30

merged 5 commits into from
Feb 9, 2020

Conversation

NickleDave
Copy link
Collaborator

@NickleDave NickleDave commented Feb 9, 2020

because Tensorflow 1.0 is deprecated.
Chose PyTorch over Tensorflow 2.0 for several reasons, see https://github.com/NickleDave/vak/pull/88

  • make vak a dependency, because:
    • we change model.py so that it subclasses vak.Model and specifies optimizer, loss, metrics
  • and add network.py module with TweetyNet class, that implements network as subclass of torch.nn.Module
  • add an initial tests/ module

- no longer import TweetyNet class
- now the only REQUIRED dependency is torch and attrs (for config)
- move 'vak' and 'crowsetta' dependencies to EXTRAS['article']
  + i.e. you can make an instance of the model with just torch,
    but if you want to recreate the article you need all the other
    libraries
- fix/add entry points
- add pytest dependency for EXTRAS['test'] in setup.py
@NickleDave NickleDave merged commit a7622be into master Feb 9, 2020
@NickleDave NickleDave deleted the make-torch-model branch February 9, 2020 16:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant