Jie Yang

Results 3 issues of Jie Yang

This PR adds `state_dict` and `load_state_dict` for current LR Schedulers. Please check it!

Platform: two different cards, 1080Ti, 2080Ti, run on 2080Ti will report it, 1080Ti is well. ![image](https://user-images.githubusercontent.com/15647890/126063310-e656a63d-e70c-4bfe-989f-6376817168ff.png) ![image](https://user-images.githubusercontent.com/15647890/126063330-04983b96-534b-4f60-949b-22b76566982b.png) Do you have any idea? Thanks!

Very wonderful work on the decompostion of NeRF. I want to try you demo. But, I have encounterd a problem when running the training script. `ModuleNotFoundError: No module named 'cosine_triline_feature_cuda'`...