search

BNL-DAQ-LDRD / NeuralCompression

3 stars 2 forks source link

A correct inference branch #14

Closed pphuangyi closed 2 years ago

pphuangyi commented 2 years ago

Hi Ray,

Here is a summary of the pull request:

  1. train.py: The script to BCAE model.
  2. inference.py: Load pre-trained encoder and run inference.
  3. The scriptors:
  4. README.md: Read-me for training, inference, the scriptors, and a to-do list.

Please let me know how I can improve them! Thank you

YHRen commented 2 years ago

Hi Ray,

Here is a summary of the pull request:

1. [`train.py`](https://github.com/BNL-DAQ-LDRD/NeuralCompression/blob/inference/train.py): The script to BCAE model.

2. [`inference.py`](https://github.com/BNL-DAQ-LDRD/NeuralCompression/blob/inference/inference.py): Load pre-trained encoder and run inference.

3. The scriptors:

   * [`neuralcompress/utils/bcae_scriptor.py`](https://github.com/BNL-DAQ-LDRD/NeuralCompression/blob/inference/neuralcompress/utils/bcae_scriptor.py): Script pre-trained bcae encoder/decoder.
   * [`neuralcompress/utils/tpc_scriptor.py`](https://github.com/BNL-DAQ-LDRD/NeuralCompression/blob/inference/neuralcompress/utils/tpc_scriptor.py): Script tensor for being used by C++.

4. [`README.md`](https://github.com/BNL-DAQ-LDRD/NeuralCompression/blob/inference/README.md): Read-me for training, inference, the scriptors, and a to-do list.

Please let me know how I can improve them! Thank you

Hi @pphuangyi , note that the way you hyperlink the files in markdown will not work in github.

YHRen commented 2 years ago

@pphuangyi Thanks a lot for these nice changes. I have merged it to main. I will delete this branch soon.