Skip to content

train error #175

@JLUForever

Description

@JLUForever

When I execute the "train.py" file, I met an error below:

2021-08-08 10:52:09.820154: W tensorflow/core/framework/op_kernel.cc:1273] OP_REQUIRES failed at matmul_op.cc:478 : Resource exhausted: OOM when allocating tensor with shape[12544,2048] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc

My GPU is a GeForce GTX 1080.

I changed the vocab_size from 32000 to 2000, but it didn't work, what should I do?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions