Spaces:
Build error
Build error
_base_ = [ | |
'./panoptic_fpn_r50_fpn_1x_predcls_psg.py', | |
] | |
model = dict( | |
relation_head=dict( | |
head_config=dict( | |
# NOTE: Evaluation type | |
use_gt_box=False, | |
use_gt_label=False, | |
), ), | |
roi_head=dict(bbox_head=dict(type='SceneGraphBBoxHead'), ), | |
) | |
evaluation = dict(interval=1, | |
metric='sgdet', | |
relation_mode=True, | |
classwise=True, | |
iou_thrs=0.5, | |
detection_method='pan_seg') | |
# Change batch size and learning rate | |
data = dict(samples_per_gpu=8, | |
# workers_per_gpu=2 | |
) | |
# Log config | |
project_name = 'openpsg' | |
expt_name = 'motifs_panoptic_fpn_r50_fpn_1x_sgdet_psg' | |
work_dir = f'./work_dirs/{expt_name}' | |
log_config = dict( | |
interval=50, | |
hooks=[ | |
dict(type='TextLoggerHook'), | |
dict( | |
type='WandbLoggerHook', | |
init_kwargs=dict( | |
project=project_name, | |
name=expt_name, | |
), | |
), | |
], | |
) | |