"projects/TensorMask/train_net.py" did not exist on "5b3792fc3ef9ab6a6f8f30634ab2e52fb0941af3"
Unverified Commit bbb4a9a1 authored by NVS Abhilash's avatar NVS Abhilash Committed by GitHub
Browse files

Add warning and note docs for scipy (#1842) (#1966)

parent b70e333a
......@@ -183,11 +183,19 @@ Inception v3
.. autofunction:: inception_v3
.. note ::
This requires `scipy` to be installed
GoogLeNet
------------
.. autofunction:: googlenet
.. note ::
This requires `scipy` to be installed
ShuffleNet v2
-------------
......
......@@ -10,6 +10,10 @@ from .utils import download_and_extract_archive, makedir_exist_ok, verify_str_ar
class Caltech101(VisionDataset):
"""`Caltech 101 <http://www.vision.caltech.edu/Image_Datasets/Caltech101/>`_ Dataset.
.. warning::
This class needs `scipy <https://docs.scipy.org/doc/>`_ to load target files from `.mat` format.
Args:
root (string): Root directory of dataset where directory
``caltech101`` exists or will be saved to if download is set to True.
......
......@@ -13,6 +13,10 @@ class SVHN(VisionDataset):
we assign the label `0` to the digit `0` to be compatible with PyTorch loss functions which
expect the class labels to be in the range `[0, C-1]`
.. warning::
This class needs `scipy <https://docs.scipy.org/doc/>`_ to load data from `.mat` format.
Args:
root (string): Root directory of dataset where directory
``SVHN`` exists.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment