Browse Source

fix(ci): fix integration_test bugs because of dataloader

GitOrigin-RevId: 63b14876cb
master
Megvii Engine Team 2 years ago
parent
commit
7790040e23
2 changed files with 6 additions and 3 deletions
  1. +5
    -3
      imperative/python/megengine/data/dataloader.py
  2. +1
    -0
      imperative/python/requires.txt

+ 5
- 3
imperative/python/megengine/data/dataloader.py View File

@@ -184,9 +184,11 @@ class DataLoader:
logger.warning(
"Each worker need to read the shared meta-data, which will be increasing the reference count."
"Copy-On-Write propety will lead to 'memory leak', the memory usage will end up being "
+ total_memory
+ " GB"
"However the current requested memory is " + current_memory + " GB"
+ str(total_memory)
+ " GB, "
"However the current requested memory is "
+ str(current_memory)
+ " GB, "
"Maybe you can request more memory or uesd np-array to save meta-data rather than List or Tuple"
)



+ 1
- 0
imperative/python/requires.txt View File

@@ -9,3 +9,4 @@ mprop
wheel
megfile>=0.0.10
filelock
psutil

Loading…
Cancel
Save