AaronJny / xyolo

一个高度封装的yolov3类库。
MIT License
35 stars 7 forks source link

unfreeze_train_epochs设置失效 #3

Closed CofeCup closed 3 years ago

CofeCup commented 3 years ago

我在 train.py 中的设置:

class MyConfig(DefaultYolo3Config):
    def __init__(self):
        super(MyConfig, self).__init__()
        ...
        self.frozen_train_epochs = 32
        self.unfreeze_train_epochs = 100

训练过程中在32轮的 frozen_train 后,unfreeze_train是显示从第 101轮开始,最终在第132轮结束。

AaronJny commented 3 years ago

确实是个bug,我晚上修一下……