nicknochnack / TFODCourse

957 stars 924 forks source link

Can we change the training in number of training steps to number of epochs? #91

Open asish-ia opened 2 years ago

asish-ia commented 2 years ago

As a newbie here, I came across two different models, from your YouTube channel. The first one was an SSD MobileNet V2 FPNLite 320x320, using tensorflow and the second one was a Yolo v5s, using Pytorch. The former trained my custom dataset using number of training steps, while the latter used number of epochs. I wish to compare the performance of both the models. Will I be able to convert the first model to work in epochs?