Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training workflow does not work on Visual Genome #7

Open
YerongLi2 opened this issue Apr 2, 2022 · 1 comment
Open

Training workflow does not work on Visual Genome #7

YerongLi2 opened this issue Apr 2, 2022 · 1 comment

Comments

@YerongLi2
Copy link

python tools/train_net_step_rel.py --dataset vg8k --cfg configs/vg8k/e2e_relcnn_VGG16_8_epochs_vg8k_y_loss_only_hubness100k.yaml --nw 8 --use_tfboard --seed 3

INFO train_net_step_rel.py: 663: Save ckpt on exception ...
INFO train_net_step_rel.py: 150: save model: Outputs/e2e_relcnn_VGG16_8_epochs_vg8k_y_loss_only_hubness100k/vg8k/Apr02-00-17-05_bock_step_with_prd_cls
_v3/ckpt/model_step0.pth
INFO train_net_step_rel.py: 665: Save ckpt done.
Traceback (most recent call last):
  File "tools/train_net_step_rel.py", line 604, in main
    net_outputs = maskRCNN(**input_data)
  File "/home/wentaoy2/anaconda3/envs/vr/lib/python3.7/site-packages/torch/nn/modules/module.py", line 477, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/wentaoy2/LTVRR/lib/nn/parallel/data_parallel.py", line 108, in forward
    outputs = [self.module(*inputs[0], **kwargs[0])]
  File "/home/wentaoy2/anaconda3/envs/vr/lib/python3.7/site-packages/torch/nn/modules/module.py", line 477, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/wentaoy2/LTVRR/lib/modeling/model_builder_rel.py", line 513, in forward
    return self._forward(data, im_info, dataset_name, roidb, use_gt_labels, include_feat, **rpn_kwargs)
  File "/home/wentaoy2/LTVRR/lib/modeling/model_builder_rel.py", line 553, in _forward
    assert len(roidb) == 1

AssertionError

@aniket-agarwal1999
Copy link
Collaborator

Hi,
I am not sure why this could be occurring. I tried to reproduce this by a completely new setup but did not get this error. Can I once know if you have properly compiled the Detectron submodule along with all the other instructions on the README?

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

No branches or pull requests

2 participants