Skip to content
Snippets Groups Projects
Unverified Commit e7dd1a56 authored by Neta Zmora's avatar Neta Zmora Committed by GitHub
Browse files

model_summaries.py: remove "experimental" warning

This is now tested and supported when using CNNs and PyTorch 0.4
parent b48908c3
No related branches found
No related tags found
No related merge requests found
...@@ -16,9 +16,8 @@ ...@@ -16,9 +16,8 @@
"""This module provides various model summary information. """This module provides various model summary information.
This code is currently in "experimental" state, as we wait for the next release This code is proven to work on CNN image classification models using PyTorch 04.
of PyTorch with stable support for the JIT tracer functionality we employ in this RNNs are currently not working well.
code (it was built with a 4.x master branch).
""" """
import re import re
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment