训练 slowfast模型的记录

SlowFast网络

Slow分支用于捕获语义信息,而Fast分支用于捕获运动信息,这和双流(two-stream)网络的思路类似。

 

训练 slowfast模型的记录_第1张图片

 

查看GPU的使用状态

nvidia-smi -l

训练 slowfast模型的记录_第2张图片

 

 

RuntimeError: Expected object of scalar type Long but got scalar type Int for argument #2 'target'

labels = labels.to('cuda', dtype=torch.int64)

labels = labels.cuda()
labels = labels.to('cuda', dtype=torch.int64)
outputs = model(i

你可能感兴趣的:(Deep,Learning)