YAO/YAO-optimizer
1
0
Fork 0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-15 09:06:43 +00:00
Code Issues Releases Wiki Activity

add files

Browse Source
This commit is contained in:
Newnius
2020-04-29 18:39:13 +08:00
parent e1c0c452ad
commit a8a1a06d75
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@@ -3,7 +3,7 @@ import numpy as np
import os
from sklearn.model_selection import train_test_split
frame = "pytorch"
frame = "tensorflow"
if frame == "pytorch":
from model.model_pytorch import train, predict
elif frame == "keras":