DynaDojo / dynadojo

An Extensible Benchmarking Platform for Scalable Dynamical System Identification
https://dynadojo.github.io/dynadojo/
MIT License
14 stars 11 forks source link

improve: migrate from tensorflow to pytorch #42

Open carynbear opened 3 months ago

carynbear commented 3 months ago

Tensorflow is used in:

Rather than use multiple ML frameworks, we want to simplify and use either torch or tensorflow. I think torch is the better option because of PyTorch's dynamic computation graph, ease of debugging, and its increasing popularity in the research community.

Once refactored, please test and make sure the demos are working with similar results. And then remove tensorflow packages from pyproject.toml and installation from Docker.

Rebuild and publish Docker images accordingly.

carynbear commented 1 month ago

src/dynadojo/baselines/dnn.py --> refactored to use torch. tensorflow version is now in src/dynadojo/baselines/dnn_keras.py.

carynbear commented 1 month ago
  1. Let's just remove these systems and algorithms from the code and documentation entirely.
  1. Remove tensorflow from the pyproject.toml

@carynbear will need to handle docker rebuilding, ignore that for now.