Commit 3e4fe139 authored by Hemerson Pistori's avatar Hemerson Pistori
Browse files

Ajustando uma questão com a configuração da GPU

parent 36f7804d
......@@ -71,6 +71,10 @@ Driver nvidia: 450.66
ann_file=data_root + 'test/_annotations.coco.json', # Diretorio para o arquivo de anotacoes de teste
img_prefix=data_root + 'test/', # Diretorio para a pasta com todas as imagens de teste
pipeline=test_pipeline))
IMPORTANTE, no caso de problemas com uma única classe, se tiver dando erro, tente repetir a
classe no comando classes=(...) para os 3 conjuntos. Por exemplo: classes=("ovo","ovo") ao
invés de classes=("ovo"). Não sei bem o motivo, mas para um teste que eu fiz funcionou
```
- Quaisquer arquivo de configuração de rede que sera usado, dependendo do arquivo.
É recomendado utilizar a [documentação do MMDetetion](https://mmdetection.readthedocs.io/en/latest/tutorials/config.html) para aprender exatamente o que cada arquivo base e variável modifica na rede. Em geral, a maioria das redes vão ser compostas de 4 componentes, mas podem ter menos ou nenhum componente base pronto:
......@@ -175,7 +179,7 @@ Feche a janela da imagem para ver a próxima
cd ../run
sh treina.sh
```
- Caso deseje rodar multiplas redes em paralelo (caso tenha acesso a multiplas GPUs), rode a primeira rede, modifique a variável gpu-ids a seguir para a outra GPU que deseje usar, modifique `./run/arquitetura.txt`, e rode a nova rede.
- Caso deseje rodar multiplas redes em paralelo (caso tenha acesso a multiplas GPUs), rode a primeira rede, modifique a variável gpu-ids (no arquivo run/treina.sh ou no run/testa.sh) a seguir para a outra GPU que deseje usar, modifique `./run/arquitetura.txt`, e rode a nova rede.
- Cuidado: Segundo alguns experimentos, certas redes não são capazes de rodar em paralelo com outras redes, dividindo poder de processamento, podendo acarretar a problemas como a perda (*loss*) alcancando NaN e nao treinando corretamente, entre outros problemas.
8. Para rodar o teste:
```shell
......
......@@ -12,7 +12,7 @@ echo 'Executando Teste' $config
cd ../mmdetection
python tools/test.py $config $pesos --show-dir ../results --out ../results/results.pkl #--eval bbox
python tools/test.py $config $pesos --gpu-ids 0 --show-dir ../results --out ../results/results.pkl #--eval bbox
cd ../run
......
......@@ -18,7 +18,7 @@ bash ../utils/moveConfigs.sh
echo 'Executando ' $config
cd ../mmdetection
python tools/train.py $config --gpu-ids 1
python tools/train.py $config --gpu-ids 0
echo 'Salvou pesos aprendidos em ' $pesos
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment