提交 ce674856 编写于 作者: J Jinhui Yuan 提交者: GitHub

enlarge the cudnn buf to 4GB (#1269)



Former-commit-id: 28f981eb
上级 9fda43bf
......@@ -679,7 +679,7 @@ message OperatorConf {
optional bool trainable = 3 [default = true];
optional DeviceType device_type = 4 [default = kInvalidDevice];
optional bool enable_cudnn = 5;
optional int64 cudnn_buf_limit_mbyte = 6 [default = 1024]; // 1GByte
optional int64 cudnn_buf_limit_mbyte = 6 [default = 4096]; // 4GByte
oneof op_type {
// system op
DecodeOFRecordOpConf decode_ofrecord_conf = 101;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册