nbnet's Introduction
Code for Paper: G. Mai, Kai Cao, P. C. Yuen and Anil K. Jain, "On the Reconstruction of Face Images from Deep Face Templates", IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI), 2019 Please cite our paper if you find the code useful. Clone the project from the github with git clone [email protected]:csgcmai/NBNet.git Install the required packages, the usage of the conda enviornment is recommended. Run the following commands conda create -n nbnet python=2.7 anaconda pip install --upgrade pip Piror to the next step, please refer https://www.tensorflow.org/install/gpu and http://mxnet.io/install/index.html for installing the necessary packages for installing the tensorflow and mxnet with GPU support cd NBNet pip install -r requirement.txt cd src python train_of2img_mae.py --gpus 0,1,2,3
nbnet's People
nbnet's Issues
RuntimeError: simple_bind error. Arguments: data: (64, 100)
**Hi,
I am trying to run the "train_of2img_mae.py" code on Colab using mxnet_cu11.1 and CUDA 11.1.
I get the following error message: (When I try it using mxnet-cu10.1 and CUDA 10.1, the same error occurs)
Can you please advise me regarding this error?**
`WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
/usr/local/lib/python3.7/dist-packages/mxnet/module/base_module.py:55: UserWarning: You created Module with Module(..., label_names=['softmax_label']) but input with name 'softmax_label' is not found in symbol.list_arguments(). Did you mean one of:
data
warnings.warn(msg)
/usr/local/lib/python3.7/dist-packages/mxnet/module/base_module.py:67: UserWarning: Data provided by label_shapes don't match names specified by label_names ([] vs. ['softmax_label'])
warnings.warn(msg)
Traceback (most recent call last):
File "/usr/local/lib/python3.7/dist-packages/mxnet/symbol/symbol.py", line 1938, in simple_bind
ctypes.byref(exe_handle)))
File "/usr/local/lib/python3.7/dist-packages/mxnet/base.py", line 246, in check_call
raise get_last_ffi_error()
mxnet.base.MXNetError: Traceback (most recent call last):
File "../src/engine/./../common/cuda_utils.h", line 394
CUDA: Check failed: e == cudaSuccess || e == cudaErrorCudartUnloading: invalid device ordinal
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "train_of2img_mae.py", line 159, in
main(args)
File "train_of2img_mae.py", line 64, in main
model_prefix=args.face_gen_prefix, load_epoch=args.face_gen_epoch)
File "/content/drive/MyDrive/NBNet/src/util/util.py", line 63, in img_generator
model_trained.bind(data_shapes=[('data',(batch_size,100))],label_shapes=None, for_training=False)
File "/usr/local/lib/python3.7/dist-packages/mxnet/module/module.py", line 429, in bind
state_names=self._state_names)
File "/usr/local/lib/python3.7/dist-packages/mxnet/module/executor_group.py", line 280, in init
self.bind_exec(data_shapes, label_shapes, shared_group)
File "/usr/local/lib/python3.7/dist-packages/mxnet/module/executor_group.py", line 384, in bind_exec
shared_group))
File "/usr/local/lib/python3.7/dist-packages/mxnet/module/executor_group.py", line 678, in _bind_ith_exec
shared_buffer=shared_data_arrays, **input_shapes)
File "/usr/local/lib/python3.7/dist-packages/mxnet/symbol/symbol.py", line 1944, in simple_bind
raise RuntimeError(error_msg)
RuntimeError: simple_bind error. Arguments:
data: (64, 100)
Traceback (most recent call last):
File "../src/engine/./../common/cuda_utils.h", line 394
CUDA: Check failed: e == cudaSuccess || e == cudaErrorCudartUnloading: invalid device ordinal`
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
๐ Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. ๐๐๐
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google โค๏ธ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.