dgl-32x1-cosine_300e-partnet_chair.py

import os
from yacs.config import CfgNode as CN
from multi_part_assembly.utils import merge_cfg

_base_ = {
    'exp': '../_base_/default_exp.py',
    'data': '../_base_/datasets/partnet/partnet_chair.py',
    'optimizer': '../_base_/schedules/adam_cosine.py',
    'model': '../_base_/models/dgl.py',
    'loss': '../_base_/models/loss/semantic_loss.py',
}

# Miscellaneous configs
_C = CN()

_C.exp = CN()
_C.exp.num_epochs = 300  # longer training time
_C.exp.val_every = 5  # DGL training is very slow

_C.data = CN()
_C.data.data_keys = ('part_ids', 'match_ids', 'contact_points', 'valid_matrix')


def get_cfg_defaults():
    base_cfg = _C.clone()
    cfg = merge_cfg(base_cfg, os.path.dirname(__file__), _base_)
    return cfg