Hugging Face's logo

Spaces:
crossentropy-ai
/
rlcube
Sleeping

App Files Community
imwithye commited on
Commit
03ff230
·
1 Parent(s): 970c966

deploy

Browse files
Files changed (4) hide show
  1. .gitattributes +1 -0
  2. rlcube/models/.gitignore +1 -0
  3. rlcube/models/model_final.pth +3 -0
  4. rlcube/rlcube/models/search.py +1 -1
.gitattributes CHANGED
@@ -1,2 +1,3 @@
1
  *.blend filter=lfs diff=lfs merge=lfs -text
2
  *.pt filter=lfs diff=lfs merge=lfs -text
 
 
1
  *.blend filter=lfs diff=lfs merge=lfs -text
2
  *.pt filter=lfs diff=lfs merge=lfs -text
3
+ *.pth filter=lfs diff=lfs merge=lfs -text
rlcube/models/.gitignore CHANGED
@@ -1,2 +1,3 @@
1
  *
2
  !.gitignore
 
 
1
  *
2
  !.gitignore
3
+ !model_final.pth
rlcube/models/model_final.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6500e9d26531992fd55179756df2152f1ad60958b0c877c9dbb91dcdc437ddc7
3
+ size 17428981
rlcube/rlcube/models/search.py CHANGED
@@ -5,7 +5,7 @@ from rlcube.envs.cube2 import Cube2Env
5
  from tqdm import tqdm
6
 
7
  net = DNN()
8
- net.load("models/model_best.pth")
9
  net.eval()
10
 
11
 
 
5
  from tqdm import tqdm
6
 
7
  net = DNN()
8
+ net.load("models/model_final.pth")
9
  net.eval()
10
 
11