Переобучить модель U2net ⇐ Python

[Гость]

Я пытаюсь переобучить модель u2net. Но по нему нет хорошо написанной документации. Я хочу обучить модель, используя набор данных из 50 000.

epoch_num = 100000 пакетный_размер_поезд = 2 пакетный_размер_знач = 1 поезд_номер = 0 знач_номер = 0 Мне не известны эти цифры; может кто-нибудь помочь мне с этим.