How could i train my network on GPU?
    2 visualizaciones (últimos 30 días)
  
       Mostrar comentarios más antiguos
    
    Hind Haboubi
 el 24 de Abr. de 2021
  
    
    
    
    
    Comentada: Hind Haboubi
 el 28 de Abr. de 2021
            Hello guys, this is my code and i don't know how to train it on single GPU? Can you help me please. i have download GPU Coder ut i don't know what to do with it? 
imageSize = [224 224 3];
load Damagevehicle
numClasses = 1;
anchorBoxes = [
    43 59
    18 22
    23 29
    84 109
];
base = resnet50;
inputlayer = base.Layers(1);
middle = base.Layers(2:174);
finallayer = base.Layers(174:end);
baseNetwork = [inputlayer
    middle
    finallayer];
featureLayer = 'activation_40_relu';
lgraph = yolov2Layers(imageSize,numClasses,anchorBoxes,base,featureLayer);
options = trainingOptions('sgdm','MiniBatchSize',128,'InitialLearnRate',1e-3,'MaxEpochs',1,'CheckpointPath',tempdir,...
    'Shuffle','every-epoch','ExecutionEnvironment','auto');
damageDataset = ddd;
[detector,info] = trainYOLOv2ObjectDetector(damageDataset,lgraph,options);
0 comentarios
Respuesta aceptada
  Tarunbir Gambhir
    
 el 27 de Abr. de 2021
        
      Editada: Tarunbir Gambhir
    
 el 27 de Abr. de 2021
  
      For training the network on a single particular GPU, you can select that GPU before you start the training and set the 'ExecutionEnvironment' as 'gpu' in trainingOptions(). You can refer this documentation for further information.
Más respuestas (0)
Ver también
Categorías
				Más información sobre Parallel and Cloud en Help Center y File Exchange.
			
	Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!