PyTorch code for paper: Few-shot multiscene fault diagnosis of rolling bearing under compound variable working conditions

For PU dataset experiments, please download PU dataset

meta transfer learning 1 shot:

python main_meta_transfer.py -s 1

you can change -b parameter based on your GPU memory.

If you use this code in your research, please use the following entry for citation.

Wang, Sihan, Dazhi Wang, Deshan Kong, Wenhui Li, Jiaxing Wang, and Huanjie Wang. "Few‐shot multiscene fault diagnosis of rolling bearing under compound variable working conditions." IET Control Theory & Applications 16, no. 14 (2022): 1405-1416.