[Instance Segmentation] Train code
def train_model train_set = Dataset(train_dataset, ...) train_generator = torch.utils.data.DataLoader(train_set, batch_size=1, shuffle=True, num_workers=4) # val_set = # val_generator = # Train optimizer = optim.SGD(params_dict, lr, momentum) for epoch in range(self.epoch+1, epochs+1): record log # Training loss, loss_rpn_class, loss_rpn_bbox, loss_mrcnn_class, los_mrcnn_bbox, loss_mrcnn_mask = ..
2019. 7. 25.
