5 Commits

Author	SHA1	Message	Date
Michael Hsu	16691c00f8	Let sample.py can generate caption for gray image. ... The sample.py cannot be used to generate caption for gray image.In the data_loader.py, all of the images have been converted to 'RGB' format. But when generating captions using sample.py for single image, this conversion is missed. So for gray image, it will have such RuntimeError: Given groups=1, weight of size [64, 3, 7, 7], expected input[1, 1, 224, 224] to have 3 channels, but got 1 channels instead.	2019-02-24 10:02:07 +08:00
yunjey	78c6afe681	Update tutorials for pytorch 0.4.0	2018-05-10 17:52:01 +09:00
yunjey	15488e0db1	delete	2018-05-10 17:47:00 +09:00
yunjey	ab5ac31dae	edit image captioning code	2017-05-29 15:07:40 +09:00
yunjey	c548e2ae9f	tutorial updated	2017-05-28 20:06:40 +09:00