Skip to content

Task killed when training for large zarr file #2

Description

@DZGong

I was trying to train the model with a data set of 40x4096x4096x3 (NWHC), but the process was always killed, as shown in the following snapshot. This doesn't happen if I switch to a smaller data set (10x4096x4096) and the train goes well.

Screenshot 2023-09-13 at 3 00 36 PM

The dataset was originally a single tiff file and was transformed into a zarr file by using function zarr.convinient.save(). The data set was then split into train, val, and test by using the code multiscale_zarr_data_generator.py. Then started training by run.py.

The computation system I used has 68G(?) CPU memory and 16G GPU memory as shown in the following snapshot:
Screenshot 2023-09-13 at 9 59 50 PM

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