Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

new_model.py #19

Open
chenyafang opened this issue Apr 21, 2019 · 6 comments
Open

new_model.py #19

chenyafang opened this issue Apr 21, 2019 · 6 comments

Comments

@chenyafang
Copy link

new_model.py和model.py的区别是什么,是不是运行new_model.py就是训练模型了,就不用再运行model.py了。谢谢!

@shartoo
Copy link
Owner

shartoo commented Apr 22, 2019

model.py的代码好像是有点问题的,建议使用new_model.py

@chenyafang
Copy link
Author

请问python new_model.py是训练模型,运行new_model.py需要很大的内存吗,我的怎么总是出现ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[1000,64,6,20,20]
[[Node: arch-1/MaxPool3D = MaxPool3DT=DT_FLOAT, data_format="NDHWC", ksize=[1, 1, 1, 1, 1], padding="SAME", strides=[1, 1, 1, 1, 1], _device="/job:localhost/replica:0/task:0/device:GPU:0"]] 是内存不够吗?但是我已经把batch_size调小了好多,还是出现这种错误。请问该怎样解决,谢谢!

@shartoo
Copy link
Owner

shartoo commented Apr 22, 2019

batch_size 建议为32或者16吧。

@chenyafang
Copy link
Author

好的,谢谢。

@chenyafang
Copy link
Author

每次训练的pos_mun+neg_num=320,比如,第一轮训练pos_num=2,neg_num=318,为什么总和是320呢?谢谢!

@shartoo
Copy link
Owner

shartoo commented Apr 22, 2019

好像没有这样的指定啊?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants