from tqdm import tqdm

epochs = 10
epoch_bar = tqdm(range(epochs))
count =0
for _ in epoch_bar:
    count = count+1
    print("count = {}".format(count))
    print(_)

每次就是一个epoch