Fix running on Edge Impulse?

Author: janjongboom

dependencies/install_cuda.sh

@@ -28,7 +28,6 @@ cp /tmp/cuda-nvcc/usr/local/cuda-${cuda_short}/bin/ptxas \
 cp /tmp/cuda-nvvm/usr/local/cuda-${cuda_short}/nvvm/libdevice/libdevice.10.bc \
     /usr/local/cuda-${cuda_short}/nvvm/libdevice/
 
-rm -f /usr/lib/x86_64-linux-gnu/libcudnn_engines_precompiled.so*
 rm -rf /tmp/cuda-nvvm /tmp/cuda-nvcc \
     cuda-nvvm-${cuda_package_version}_*.deb cuda-nvcc-${cuda_package_version}_*.deb \
     /usr/share/doc/* /usr/share/man/* /usr/share/locale/*

shared/training.py

@@ -16,8 +16,7 @@ def get_callbacks(dir_path, is_enterprise_project, max_training_time_s, max_gpu_
 
     if enable_tensorboard:
         tb_callback = tf.keras.callbacks.TensorBoard(log_dir=os.path.join(dir_path, 'tensorboard_logs'),
-                                                     # Profile batches 1-100
-                                                     profile_batch=(1,101))
+                                                     profile_batch=0)
         callbacks.append(tb_callback)
 
     return callbacks