将torch t7模型转换为pytorch模型

我有一个torch t7模型,我想将其转换为pytorch模型。我使用了这种方法:

model = load_lua('xxx.t7', unknown_classes=True)

但是,我得到了以下错误:

AttributeError: type object 'torch.cuda.FloatStorage' has no attribute 'from_buffer'

有什么想法如何解决吗?

点赞