Description
if it is possible to provide the checkpoint?
and in the codes i find that due to the config in :
def add_model_specific_args(parent_parser):
parser = parent_parser.add_argument_group('HiVT')
parser.add_argument('--historical_steps', type=int, default=20)
parser.add_argument('--future_steps', type=int, default=30)
parser.add_argument('--num_modes', type=int, default=6)
parser.add_argument('--rotate', type=bool, default=True)
parser.add_argument('--node_dim', type=int, default=2)
parser.add_argument('--edge_dim', type=int, default=2)
parser.add_argument('--embed_dim', type=int, required=True)
parser.add_argument('--num_heads', type=int, default=8)
parser.add_argument('--dropout', type=float, default=0.1)
parser.add_argument('--num_temporal_layers', type=int, default=4)
parser.add_argument('--num_global_layers', type=int, default=3)
parser.add_argument('--local_radius', type=float, default=50)
parser.add_argument('--parallel', type=bool, default=False)
parser.add_argument('--lr', type=float, default=5e-4)
parser.add_argument('--weight_decay', type=float, default=1e-4)
parser.add_argument('--T_max', type=int, default=64)
return parent_parser
the refine_radius is always 50 which should be -1 according to the paper