Deep Neural Network Prediction

Step 1. Genernate random parameters and Run them sequentially : $ python3 collect_data.py -gp -ep -pp -pl pooling -num 10 -shuffle -d 1080ti

Step 2. Data Timeline parser : $ python3 preprocess_data.py -pt -pl pooling -d 1080ti

Step 3. Combine all raw data : $ python3 preprocess_data.py -c -pl pooling -d 1080ti

Step 4. Split raw data to train and test data as performance prediction inputs $ python3 preprocess_data.py -sp -pl pooling -d 1080ti

Step 5. Train Model :

python3 train_model.py -ftf ./utils/Feature_Target/conv_pre.json -log2file 1 -e 1000 -st 400 -sg 0.5 -lf maple -n perfnetA -pd 1080ti -pl convolution -psl pre
    
python3 train_model.py -ftf ./utils/Feature_Target/conv_exe.json -log2file 1 -e 1000 -st 400 -sg 0.5 -lf maple -n perfnetA -pd 1080ti -pl convolution -psl exe
        
python3 train_model.py -ftf ./utils/Feature_Target/conv_post.json -log2file 1 -e 1000 -st 400 -sg 0.5 -lf maple -n perfnetA -pd 1080ti -pl convolution -psl post

Step 6. Generate model csv file :

python3 verify_model.py -gmc --model lenet -b 1

Step 7. Predict model csv file :

python3 verify_model.py -pdm -n perfnetA -d 1080ti -l -eva -lf maple --model lenet -b 1

Or you can use script to predict all models with all batches:

python3 predict_script.py -n perfnetA -d 1080ti -lf maple

Predict the latency time of the deep learning models

Related tags

Overview

Deep Neural Network Prediction

Owner

QAQ

Unofficial PyTorch code for BasicVSR

PyTorch implementation of Self-supervised Contrastive Regularization for DG (SelfReg)

Sync2Gen Code for ICCV 2021 paper: Scene Synthesis via Uncertainty-Driven Attribute Synchronization

A rough implementation of the paper "A Steering Algorithm for Redirected Walking Using Reinforcement Learning"

Reinforcement Learning via Supervised Learning

Code for the paper Task Agnostic Morphology Evolution.

Unified Pre-training for Self-Supervised Learning and Supervised Learning for ASR

Energy consumption estimation utilities for Jetson-based platforms

Code of the lileonardo team for the 2021 Emotion and Theme Recognition in Music task of MediaEval 2021

这是一个deeplabv3-plus-pytorch的源码，可以用于训练自己的模型。

This is the official PyTorch implementation of the CVPR 2020 paper "TransMoMo: Invariance-Driven Unsupervised Video Motion Retargeting".

CvT-ASSD: Convolutional vision-Transformerbased Attentive Single Shot MultiBox Detector (ICTAI 2021 CCF-C 会议)The 33rd IEEE International Conference on Tools with Artificial Intelligence

meProp: Sparsified Back Propagation for Accelerated Deep Learning (ICML 2017)

Temporal Segment Networks (TSN) in PyTorch

WiFi-based Multi-task Sensing

Code for the Lovász-Softmax loss (CVPR 2018)

HAR-stacked-residual-bidir-LSTMs - Deep stacked residual bidirectional LSTMs for HAR

Python implementation of "Single Image Haze Removal Using Dark Channel Prior"

The official implementation of "Rethink Dilated Convolution for Real-time Semantic Segmentation"

Implementation for the "Surface Reconstruction from 3D Line Segments" paper.