import math
import torch
import torch.nn as nn
import os
def file_name_walk(file_dir):
for root, dirs, files in os.walk(file_dir):
# print(root, root) # 当前目录路径
print(dirs, dirs) # 当前路径下所有子目录
print(files, files) # 当前路径下所有非目录子文件
file_name_walk(/home/kesci/input
1