James Zhao issues

Repositories
Issues
Comments

Results 4 issues of


                                            James Zhao

Add support for MPS on MacOS with ARM chips

Support MPS on MacOS with ARM chips. Also allow CPU fallback when GPU and MPS are not available.

add support for MPS on MacOS with ARM chips (Apple Silicon)

Tested on macOS Sonama 14.6.1 wit M3 Max.

Support inference on MacOS with MPS or CPU and add face detection script

- inference on MacOS with MPS: ``` # get reconstruction results of example images python test.py --name= --epoch=20 --img_folder=./datasets/examples --device mps --renderer_type face3d # no visualization python test.py --name= --epoch=20...

support 1-d CNN and support any device type available in pytorch

- support 1-d CNN (Conv1d, MaxPool1d, AvgPool1d); also support normalization layers, linear layers, dropout layers, considering these layers as `pointwise_operations` which will not change `jump` and `receptive_field` - support any...