YAO/YAO-optimizer
1
0
Fork 0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-13 00:16:44 +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
.idea/workspace.xml generated
View File

@@ -134,7 +134,7 @@
<component name="PropertiesComponent"> <component name="PropertiesComponent">
<property name="WebServerToolWindowFactoryState" value="false" /> <property name="WebServerToolWindowFactoryState" value="false" />
<property name="aspect.path.notification.shown" value="true" /> <property name="aspect.path.notification.shown" value="true" />
<property name="com.android.tools.idea.instantapp.provision.ProvisionBeforeRunTaskProvider.myTimeStamp" value="1588156664537" /> <property name="com.android.tools.idea.instantapp.provision.ProvisionBeforeRunTaskProvider.myTimeStamp" value="1588156719300" />
<property name="go.gopath.indexing.explicitly.defined" value="true" /> <property name="go.gopath.indexing.explicitly.defined" value="true" />
<property name="nodejs_interpreter_path.stuck_in_default_project" value="undefined stuck path" /> <property name="nodejs_interpreter_path.stuck_in_default_project" value="undefined stuck path" />
<property name="nodejs_npm_path_reset_for_default_project" value="true" /> <property name="nodejs_npm_path_reset_for_default_project" value="true" />

2
main.py
View File

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