hecui

How to train?

1

Hello, when I run stage1.py, it appears: [GpuDevice(id=0, process_index=0), GpuDevice(id=1, process_index=0), GpuDevice(id=2, process_index=0), GpuDevice(id=3, process_index=0), GpuDevice(id=4, process_index=0), GpuDevice(id=5, process_index=0), GpuDevice(id=6, process_index=0), GpuDevice(id=7, process_index=0)] 2023-12-26 10:10:17.684365: W external/org_tensorflow/tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc:73] Can't find libdevice...

如题，请问如何使用自己的数据集训练 liif 模型？非常感激

As shown in the title, I want to speed up my t2v model, how should I do? Thanks a lot!