Skip to content

FduJyy/EHDvRBN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Setup

First setup the modified mxnet.

Use cmake to build the project. Make sure to install all the dependencies described here.

Adjust settings in cmake (build-type Release or Debug, configure CUDA, OpenBLAS or Atlas, OpenCV, OpenMP etc.)

$ cd mxnet
$ mkdir build/Release && cd build/Release
$ cmake ../../
$ ccmake . 
$ make -j `nproc`

Build the MXNet Python binding

$ cd mxnet/python
$ pip install --upgrade pip
$ pip install -e .

Train on ICCAD contest

First run the training script and store the best performed checkpoint.

cd my
python train.py

Then modify common/data.py and train.py to finetune the model using dataset with biased labels.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published