From e7dd1a56b4ac2b71f5e19c7ba71a34fb8e6b2f47 Mon Sep 17 00:00:00 2001 From: Neta Zmora <31280975+nzmora@users.noreply.github.com> Date: Mon, 16 Jul 2018 00:12:31 +0300 Subject: [PATCH] model_summaries.py: remove "experimental" warning This is now tested and supported when using CNNs and PyTorch 0.4 --- apputils/model_summaries.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/apputils/model_summaries.py b/apputils/model_summaries.py index 1faa8be..c09a155 100755 --- a/apputils/model_summaries.py +++ b/apputils/model_summaries.py @@ -16,9 +16,8 @@ """This module provides various model summary information. -This code is currently in "experimental" state, as we wait for the next release -of PyTorch with stable support for the JIT tracer functionality we employ in this -code (it was built with a 4.x master branch). +This code is proven to work on CNN image classification models using PyTorch 04. +RNNs are currently not working well. """ import re -- GitLab