+
Skip to content

Error: Unable to allocate 0 bytes for an array with shape #105

@DenkingOfficial

Description

@DenkingOfficial

I'm trying to run training process on biked dataset with command:
python train.py --outdir ./results --snap=10 --data=./datasets/biked --augpipe=bgcfnc --res=512

But I'm getting this error:

np_resource = np.dtype([("resource", np.ubyte, 1)]) Traceback (most recent call last): File "train.py", line 561, in <module> main() File "train.py", line 553, in main run_training(**vars(args)) File "train.py", line 416, in run_training run_desc, training_options = setup_training_options(**hyperparam_options) File "train.py", line 105, in setup_training_options dataset_obj = dataset.load_dataset(**args.train_dataset_args) # try to load the data and see what comes out File "C:\Temporary Software\stylegan2-ada\training\dataset.py", line 231, in load_dataset mirror_augment=mirror_augment, repeat=repeat, shuffle=shuffle) File "C:\Temporary Software\stylegan2-ada\training\dataset.py", line 114, in __init__ self._tf_labels_var = tflib.create_var_with_large_initial_value(self._np_labels, name='labels_var') File "C:\Temporary Software\stylegan2-ada\dnnlib\tflib\tfutil.py", line 234, in create_var_with_large_initial_value zeros = tf.zeros(initial_value.shape, initial_value.dtype) File "C:\Users\Andrew\anaconda3\envs\Stylegan\lib\site-packages\tensorflow\python\ops\array_ops.py", line 1871, in zeros output = _constant_if_small(zero, shape, dtype, name) File "C:\Users\Andrew\anaconda3\envs\Stylegan\lib\site-packages\tensorflow\python\ops\array_ops.py", line 1829, in _constant_if_small return constant(value, shape=shape, dtype=dtype, name=name) File "C:\Users\Andrew\anaconda3\envs\Stylegan\lib\site-packages\tensorflow\python\framework\constant_op.py", line 246, in constant allow_broadcast=True) File "C:\Users\Andrew\anaconda3\envs\Stylegan\lib\site-packages\tensorflow\python\framework\constant_op.py", line 284, in _constant_impl allow_broadcast=allow_broadcast)) File "C:\Users\Andrew\anaconda3\envs\Stylegan\lib\site-packages\tensorflow\python\framework\tensor_util.py", line 464, in make_tensor_proto nparray = np.empty(shape, dtype=np_dt) numpy.core._exceptions.MemoryError: Unable to allocate 0 bytes for an array with shape (1073741824, 0) and data type float32

Also I'm getting this error with every dataset that I'm trying to train
If someone need info about my environment I'll provide it

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions

      点击 这是indexloc提供的php浏览器服务,不要输入任何密码和下载