Comments (3)
According to the error message, memory size of your GPU device seems to be insufficient for the default batch size. Can you try with --batchsize
option?
$ python train_model.py ../../data/dataSplit ../../data/processedBuildingLabels/3band ../../data/buildingMaskImages --batchsize 8
Because the default batch size is set to 16, required memory size will decrease by half with the example above.
from spacenet_building_detection.
Continues the same erros, these are the data of my computer, what minimal memory should I have? or how could this problem be solved
*-core
descripción: Placa base
producto: H310M-HDV
fabricante: ASRock
id físico: 0
serie: M80-B5014702592
*-memory
descripción: Memoria de sistema
id físico: 9
ranura: Placa de sistema o placa base
tamaño: 8GiB
*-bank:0
descripción: DIMM DDR4 Síncrono 2667 MHz (0,4 ns)
producto: BLS8G4D26BFSBK.8FD
fabricante: 859B
id físico: 0
serie: A91EB06E
ranura: ChannelA-DIMM0
tamaño: 8GiB
anchura: 64 bits
reloj: 2667MHz (0.4ns)
*-cpu
descripción: CPU
producto: Intel(R) Core(TM) i5-8400 CPU @ 2.80GHz
fabricante: Intel Corp.
id físico: 15
información del bus: cpu@0
versión: Intel(R) Core(TM) i5-8400 CPU @ 2.80GHz
serie: To Be Filled By O.E.M.
ranura: CPUSocket
tamaño: 1548MHz
capacidad: 4005MHz
anchura: 64 bits
reloj: 100MHz
*-display
descripción: VGA compatible controller
producto: GM107GL [Quadro K620]
fabricante: NVIDIA Corporation
id físico: 0
información del bus: pci@0000:01:00.0
versión: a2
anchura: 64 bits
reloj: 33MHz
capacidades: pm msi pciexpress vga_controller bus_master cap_list rom
configuración: driver=nvidia latency=0
recursos: irq:126 memoria:a2000000-a2ffffff memoria:90000000-9fffffff memoria:a0000000-a1ffffff ioport:4000(size=128) memoria:c0000-dffff
*-multimedia
descripción: Audio device
producto: NVIDIA Corporation
fabricante: NVIDIA Corporation
id físico: 0.1
información del bus: pci@0000:01:00.1
versión: a1
anchura: 32 bits
reloj: 33MHz
capacidades: pm msi pciexpress bus_master cap_list
configuración: driver=snd_hda_intel latency=0
recursos: irq:17 memoria:a3080000-a3083fff
*-disk
descripción: ATA Disk
producto: KINGSTON SA400S3
id físico: 0.0.0
información del bus: scsi@0:0.0.0
nombre lógico: /dev/sda
versión: B1D1
serie: 50026B76823F300B
tamaño: 223GiB (240GB)
capacidades: partitioned partitioned:dos
configuración: ansiversion=5 logicalsectorsize=512 sectorsize=512 signature=b061fbf9
from spacenet_building_detection.
Ok, please try:
python train_model.py ../../data/dataSplit ../../data/processedBuildingLabels/3band ../../data/buildingMaskImages --batchsize 1
from spacenet_building_detection.
Related Issues (6)
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from spacenet_building_detection.