为什么模型在刚开始训练的时候比较快,但是8~9分钟后就慢很多了?

问题描述 投票:0回答:1

enter image description here

开始时间只有40分钟

enter image description here

过了一会儿,时间需要3个小时

我想也许是我的型号代码错误?

if self.spatial_frozen:
  self.spatial_backbone.eval()
  with torch.no_grad():
   x = self.spatial_backbone(x)
   #spatial_backbone dont need grad
spatial_x = rearrange(x, '(b f) d -> b f d', b=batch_size)
x = spatial_x
x = self.temporal_transformer(x)
#temporal_transformer need grad
deep-learning
1个回答
0
投票

enter image description here

这样,Image.open就被占用了好久

© www.soinside.com 2019 - 2024. All rights reserved.