SYMBOL INDEX (81 symbols across 5 files) FILE: data.py function get_paths (line 12) | def get_paths(path, name='coco', use_restval=False): class CocoDataset (line 78) | class CocoDataset(data.Dataset): method __init__ (line 81) | def __init__(self, root, json, vocab, transform=None, ids=None): method __getitem__ (line 111) | def __getitem__(self, index): method get_raw_item (line 130) | def get_raw_item(self, index): method __len__ (line 145) | def __len__(self): class FlickrDataset (line 149) | class FlickrDataset(data.Dataset): method __init__ (line 154) | def __init__(self, root, json, split, vocab, transform=None): method __getitem__ (line 165) | def __getitem__(self, index): method __len__ (line 189) | def __len__(self): class PrecompDataset (line 193) | class PrecompDataset(data.Dataset): method __init__ (line 199) | def __init__(self, data_path, data_split, vocab): method __getitem__ (line 221) | def __getitem__(self, index): method __len__ (line 238) | def __len__(self): function collate_fn (line 242) | def collate_fn(data): function get_loader_single (line 271) | def get_loader_single(data_name, split, root, json, vocab, transform, function get_precomp_loader (line 298) | def get_precomp_loader(data_path, data_split, vocab, opt, batch_size=100, function get_transform (line 311) | def get_transform(data_name, split_name, opt): function get_loaders (line 328) | def get_loaders(data_name, vocab, crop_size, batch_size, workers, opt): function get_test_loader (line 360) | def get_test_loader(split_name, data_name, vocab, crop_size, batch_size, FILE: evaluation.py class AverageMeter (line 15) | class AverageMeter(object): method __init__ (line 18) | def __init__(self): method reset (line 21) | def reset(self): method update (line 27) | def update(self, val, n=0): method __str__ (line 33) | def __str__(self): class LogCollector (line 43) | class LogCollector(object): method __init__ (line 46) | def __init__(self): method update (line 50) | def update(self, k, v, n=0): method __str__ (line 56) | def __str__(self): method tb_log (line 66) | def tb_log(self, tb_logger, prefix='', step=None): function encode_data (line 73) | def encode_data(model, data_loader, log_step=10, logging=print): function evalrank (line 123) | def evalrank(model_path, data_path=None, split='dev', fold5=False): function i2t (line 205) | def i2t(images, captions, npts=None, measure='cosine', return_ranks=False): function t2i (line 258) | def t2i(images, captions, npts=None, measure='cosine', return_ranks=False): FILE: model.py function l2norm (line 13) | def l2norm(X): function EncoderImage (line 21) | def EncoderImage(data_name, img_dim, embed_size, finetune=False, class EncoderImageFull (line 38) | class EncoderImageFull(nn.Module): method __init__ (line 40) | def __init__(self, embed_size, finetune=False, cnn_type='vgg19', method get_cnn (line 67) | def get_cnn(self, arch, pretrained): method load_state_dict (line 85) | def load_state_dict(self, state_dict): method init_weights (line 105) | def init_weights(self): method forward (line 113) | def forward(self, images): class EncoderImagePrecomp (line 134) | class EncoderImagePrecomp(nn.Module): method __init__ (line 136) | def __init__(self, img_dim, embed_size, use_abs=False, no_imgnorm=False): method init_weights (line 146) | def init_weights(self): method forward (line 154) | def forward(self, images): method load_state_dict (line 170) | def load_state_dict(self, state_dict): class EncoderText (line 185) | class EncoderText(nn.Module): method __init__ (line 187) | def __init__(self, vocab_size, word_dim, embed_size, num_layers, method init_weights (line 201) | def init_weights(self): method forward (line 204) | def forward(self, x, lengths): function cosine_sim (line 230) | def cosine_sim(im, s): function order_sim (line 236) | def order_sim(im, s): class ContrastiveLoss (line 245) | class ContrastiveLoss(nn.Module): method __init__ (line 250) | def __init__(self, margin=0, measure=False, max_violation=False): method forward (line 260) | def forward(self, im, s): class VSE (line 290) | class VSE(object): method __init__ (line 295) | def __init__(self, opt): method state_dict (line 325) | def state_dict(self): method load_state_dict (line 329) | def load_state_dict(self, state_dict): method train_start (line 333) | def train_start(self): method val_start (line 339) | def val_start(self): method forward_emb (line 345) | def forward_emb(self, images, captions, lengths, volatile=False): method forward_loss (line 360) | def forward_loss(self, img_emb, cap_emb, **kwargs): method train_emb (line 367) | def train_emb(self, images, captions, lengths, ids=None, *args): FILE: train.py function main (line 19) | def main(): function train (line 136) | def train(opt, train_loader, model, epoch, val_loader): function validate (line 187) | def validate(opt, val_loader, model): function save_checkpoint (line 220) | def save_checkpoint(state, is_best, filename='checkpoint.pth.tar', prefi... function adjust_learning_rate (line 226) | def adjust_learning_rate(opt, optimizer, epoch): function accuracy (line 234) | def accuracy(output, target, topk=(1,)): FILE: vocab.py class Vocabulary (line 22) | class Vocabulary(object): method __init__ (line 25) | def __init__(self): method add_word (line 30) | def add_word(self, word): method __call__ (line 36) | def __call__(self, word): method __len__ (line 41) | def __len__(self): function from_coco_json (line 45) | def from_coco_json(path): function from_flickr_json (line 55) | def from_flickr_json(path): function from_txt (line 64) | def from_txt(txt): function build_vocab (line 72) | def build_vocab(data_path, data_name, jsons, threshold): function main (line 107) | def main(data_path, data_name):