diff --git a/README.md b/README.md index 7e3808cac..c6dbc380c 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,6 @@ This is a **[PyTorch](https://pytorch.org) Tutorial to Image Captioning**. -This is the first in a series of tutorials I plan to write about _implementing_ cool models on your own with the amazing PyTorch library. +This is the first in [a series of tutorials](https://github.com/sgrvinod/Deep-Tutorials-for-PyTorch) I plan to write about _implementing_ cool models on your own with the amazing PyTorch library. Basic knowledge of PyTorch, convolutional and recurrent neural networks is assumed. diff --git a/datasets.py b/datasets.py index 8ca079fd5..ec6a8d11f 100644 --- a/datasets.py +++ b/datasets.py @@ -12,7 +12,6 @@ class CaptionDataset(Dataset): def __init__(self, data_folder, data_name, split, transform=None): """ - :param data_folder: folder where data files are stored :param data_name: base name of processed datasets :param split: split, one of 'TRAIN', 'VAL', or 'TEST'