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 ...@@ -183,11 +183,19 @@ Inception v3
.. autofunction:: inception_v3 .. autofunction:: inception_v3
.. note ::
This requires `scipy` to be installed
GoogLeNet GoogLeNet
------------ ------------
.. autofunction:: googlenet .. autofunction:: googlenet
.. note ::
This requires `scipy` to be installed
ShuffleNet v2 ShuffleNet v2
------------- -------------
......
...@@ -10,6 +10,10 @@ from .utils import download_and_extract_archive, makedir_exist_ok, verify_str_ar ...@@ -10,6 +10,10 @@ from .utils import download_and_extract_archive, makedir_exist_ok, verify_str_ar
class Caltech101(VisionDataset): class Caltech101(VisionDataset):
"""`Caltech 101 <http://www.vision.caltech.edu/Image_Datasets/Caltech101/>`_ Dataset. """`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: Args:
root (string): Root directory of dataset where directory root (string): Root directory of dataset where directory
``caltech101`` exists or will be saved to if download is set to True. ``caltech101`` exists or will be saved to if download is set to True.
......
...@@ -13,6 +13,10 @@ class SVHN(VisionDataset): ...@@ -13,6 +13,10 @@ class SVHN(VisionDataset):
we assign the label `0` to the digit `0` to be compatible with PyTorch loss functions which 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]` 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: Args:
root (string): Root directory of dataset where directory root (string): Root directory of dataset where directory
``SVHN`` exists. ``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