Commit 36849429 authored by Alykhan Tejani's avatar Alykhan Tejani Committed by Francisco Massa
Browse files

Fix random and center crop docs (#247)

* load densenet tensors onto cpu

* fix docs for size param in random + center crop

* revert incorrect commit
parent 5dc42237
...@@ -204,7 +204,7 @@ class CenterCrop(object): ...@@ -204,7 +204,7 @@ class CenterCrop(object):
Args: Args:
size (sequence or int): Desired output size of the crop. If size is an size (sequence or int): Desired output size of the crop. If size is an
int instead of sequence like (h, w), a square crop (size, size) is int instead of sequence like (w, h), a square crop (size, size) is
made. made.
""" """
...@@ -283,7 +283,7 @@ class RandomCrop(object): ...@@ -283,7 +283,7 @@ class RandomCrop(object):
Args: Args:
size (sequence or int): Desired output size of the crop. If size is an size (sequence or int): Desired output size of the crop. If size is an
int instead of sequence like (h, w), a square crop (size, size) is int instead of sequence like (w, h), a square crop (size, size) is
made. made.
padding (int or sequence, optional): Optional padding on each border padding (int or sequence, optional): Optional padding on each border
of the image. Default is 0, i.e no padding. If a sequence of length of the image. Default is 0, i.e no padding. If a sequence of length
......
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