文件目录
--baseline_tensorflow
--data
--model_zoo(可忽略)
data下载链接:https://awscdn.datafountain.cn/cometition_data2/Files/PengCheng2020/CSI/Hdata.mat 下载后放在data文件里即可
训练时需要修改baseline_tensorflow/Model_train.py内的文件读取和保存路径,如下:
line35 --> data_load_address = '/content/drive/My Drive/NAIC/data' # 此处改为实际文件路径 “实际路径 + /data"
line51 --> modelsave1 = '/content/drive/My Drive/NAIC/Modelsave/encoder.h5' # 此处改为 “实际路径 + /Modelsave/encoder.h5”
line55 --> modelsave2 = '/content/drive/My Drive/NAIC/Modelsave/decoder.h5' # 此处改为 “实际路径 + /Modelsave/decoder.h5”
训练结束后,导出文件Modelsave。
祝一切顺利!