NeWRF: A Deep Learning Framework for Wireless Radiation Field Reconstruction and Channel Prediction [Paper]

• Python: 3.10.9
• Pytorch: 1.13.1
• CUDA: 11.7

• simulator/ contains the simulation code. Run main.m in MATLAB to generate datasets
• train.py contains the training code. Usage:

python train.py --env [environment]

• eval.py contains the evaluation code. Usage:

python eval.py --env [environment] --ckpt [checkpoint file] 

• demo.ipynb provides visualization during the training process