about the checkpoint for the net

[Lukas88664]

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

[youngzhou1999]

Sorry for the late reply. I have been so occupied with other stuff recently. We will release the checkpoint soon.