b6443232013e8d248fe26f59630d43bc9688df06,train.py,,,#,21

Before Change

		optimizer.zero_grad()


		classification, regression, anchors, transformed_anchors = model(data["img"].cuda().float())
		if data["annot"][0, 0, 4] == -1:
			continue

After Change

model = model.resnet18(pretrained=True)

dataset_train = CocoDataset("../coco/", set_name="train2017", transform=transforms.Compose([Resizer(), ToTensor()]))
dataset_val = CocoDataset("../coco/", set_name="val2017", transform=transforms.Compose([Resizer(), ToTensor()]))

dataloader_train = DataLoader(dataset_train, batch_size=1, shuffle=True, num_workers=4, collate_fn=collater)
dataloader_val   = DataLoader(dataset_val, batch_size=1, shuffle=True, num_workers=4, collate_fn=collater)

use_gpu = True

if use_gpu:

Instances