1. Appoint GPU function
import os
os.environ["CUDA_VISIBLE_DEVICES"]="0,1"# Appoint 0,1 Number GPU Run code on
2. see GPU Occupancy rate
watch -n 1 -d nvidia-smi # Every interval 1s Refresh the display 3. Print design network parameters
print ("Total number of paramerters in networks is {} ".format (sum (x.numel () for x in model.parameters ())))
