hust-diangroup / ns3-ai

Enable the interaction between ns-3 and popular frameworks using Python, which mean you can train and test your AI algorithms in ns-3 without changing any frameworks you are using now!
GNU General Public License v2.0
220 stars 80 forks source link

add compatibility with ray and tensorflow #121

Open Kakadus opened 2 months ago

Kakadus commented 2 months ago

ray needs to pickle and unpickle the environment. This adds corresponding logic to kill and recreate the ns3 process in these cases. Also, uint is not supported by torch and tensorflow in general, so we add a notification if it is used.