Skip to content

v1.0

Latest
Compare
Choose a tag to compare
@Cyanogenoid Cyanogenoid released this 04 Aug 16:54
· 17 commits to master since this release

This is the log file containing the weights for a model that has been trained for 50 epochs (1890 iterations each) on the training set of VQA v1.
To load a model with these weights, you can run:

import torch
import model

log = torch.load('logs/2017-08-04_00:55:19.pth')
tokens = len(log['vocab']['question']) + 1

net = torch.nn.DataParallel(model.Net(tokens))
net.load_state_dict(log['weights'])