[fix] Non heuristic moves...
This commit is contained in:
@@ -0,0 +1,38 @@
|
||||
"""Entry point: python -m rl [train|export|info]"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
|
||||
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
||||
os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
|
||||
os.environ["CUDA_VISIBLE_DEVICES"] = ""
|
||||
|
||||
|
||||
def main():
|
||||
cmd = sys.argv[1] if len(sys.argv) > 1 else "train"
|
||||
|
||||
if cmd == "train":
|
||||
from .train import train
|
||||
train()
|
||||
|
||||
elif cmd == "export":
|
||||
from .export import export_tflite
|
||||
model_path = sys.argv[2] if len(sys.argv) > 2 else "rl/checkpoints/model_final.keras"
|
||||
export_tflite(model_path)
|
||||
|
||||
elif cmd == "visualize":
|
||||
from .visualize import run_visualized
|
||||
run_visualized()
|
||||
|
||||
elif cmd == "info":
|
||||
from .model import build_model, print_model_info
|
||||
model = build_model()
|
||||
print_model_info(model)
|
||||
|
||||
else:
|
||||
print(f"Unknown command: {cmd}")
|
||||
print("Usage: python -m rl [train|visualize|export|info]")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
main()
|
||||
Reference in New Issue
Block a user