Skip to content
This repository has been archived by the owner on May 7, 2019. It is now read-only.

NotImplementedError #13

Open
akira-l opened this issue Jun 4, 2018 · 2 comments
Open

NotImplementedError #13

akira-l opened this issue Jun 4, 2018 · 2 comments

Comments

@akira-l
Copy link

akira-l commented Jun 4, 2018

did anyone meet this problem?
there are some answer in PyTorch Forums said that the code should have a forward part, so how to add it in this code?

@akira-l
Copy link
Author

akira-l commented Jun 4, 2018

I run the train_coco.py directly. Maybe I need other operation ?

@akira-l
Copy link
Author

akira-l commented Jun 4, 2018

addition:
Instead of COCO dataset, I want to use retinanet in wider face. So I change the data input like this:
transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.485,0.456,0.406), (0.229,0.224,0.225))
])

trainset = ListDataset(root="./../../wider face/WIDER_train/images",
list_file="./../../wider face/wider_face_split/wider_face_train_bbx_gt.txt", train=True, transform=transform, input_size=600)
train_loader = torch.utils.data.DataLoader(trainset, batch_size=1, shuffle=True, num_workers=8, collate_fn=trainset.collate_fn)

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant