mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-03-12 12:25:32 +08:00
feat(compressors): add neural compression modules and training pipeline
This commit is contained in:
@@ -1,4 +1,15 @@
|
||||
from visualizer import app
|
||||
import argparse
|
||||
|
||||
if __name__ == "__main__":
|
||||
app.run(debug=True)
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("train")
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.train:
|
||||
from compressors import FloatCompressor, train
|
||||
|
||||
train(FloatCompressor(), 1, 32)
|
||||
else:
|
||||
from visualizer import app
|
||||
|
||||
app.run(debug=True)
|
||||
|
||||
Reference in New Issue
Block a user