我打算在每次培训中阅读2个视频。我想知道每次调用model.fit()
时是否会重新初始化模型的权重?否则将节省重量?代码如下。
length = len(videos)
num = 2
steps = length // num
for i in range(steps):
imgs, labels = many_videos_reader(videos, index=i*num, num=num)
model.fit(img,label, epochs=3, batch_size=1, initial_epoch=0)
[否,只要您拨打fit
,重量就不会重置。每当您呼叫fit
时,它们都是updated。要重置权重,您可以调用model.set_weights()
或只创建一个新模型。