efficientdet在test时遇到一些问题

在学习《手把手教物体检测——EfficientDet》过程中，我在efficientdet-test.py遇到一些问题，博主的标签是‘aircaft’和‘oiltank’（两个），我在修改成我的数据集标签后（总共有9个），出现了一下问题：copying a param with shape torch.Size([18, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([81, 64, 1, 1]。在我将我的9个标签减少至2个后，代码可正常运行。想请教在哪里可以修改这个torch.size，将其增大至我想要的大小。文章的coco.yaml、config.py我都修改过。

在checkpiont中，只有文件efficientdet-d0.pth，我尝试打开修改，但是无法打开。