YOLOv3-model-pruning/test_prune.sh

4 lines
64 B
Bash

#!bash
CARD=0,1
CUDA_VISIBLE_DEVICES=$CARD python test_prune.py