Skip to content

Commit 82d1b86

Browse files
authored
Updated information (#40)
1 parent 4154335 commit 82d1b86

3 files changed

Lines changed: 32 additions & 14 deletions

File tree

AUTHORS

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
# Names should be added to this file like so:
2+
# Name or Organization <email address>
3+
4+
# Initial version authors:
5+
Omkar Prabhu <prabhuomkar@pm.me>
6+
Markus Fleischhacker <markus.fleischhacker28@gmail.com>
7+
8+
# Partial list of contributors:

README.md

Lines changed: 24 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,15 @@
1-
<p align="center"><img width="50%" src="images/pytorch_logo.svg" /></p>
1+
<h1 align="center">
2+
<img src="images/pytorch_cpp.png" width="50%">
3+
</h1>
4+
<p align="center">
5+
C++ Implementation of PyTorch Tutorials for Deep Learning Researchers
6+
<br />
7+
<img src="https://img.shields.io/travis/prabhuomkar/pytorch-cpp">
8+
<img src="https://img.shields.io/github/license/prabhuomkar/pytorch-cpp">
9+
<img src="https://img.shields.io/badge/libtorch-1.4-ee4c2c">
10+
<img src="https://img.shields.io/badge/cmake-3.14-064f8d">
11+
</p>
212

3-
--------------------------------------------------------------------------------
4-
![MIT License](https://img.shields.io/github/license/prabhuomkar/pytorch-cpp)
5-
![C++ PyTorch](https://img.shields.io/badge/c%2B%2B-pytorch-orange)
613

714
| OS (Compiler)\\libtorch | 1.4 | 1.5 | nightly |
815
| :---------------------: | :---------------------------------------------------------------------------------------------------: | :-: | :-----: |
@@ -18,18 +25,27 @@ This repository provides tutorial code in C++ for deep learning researchers to l
1825
**Python Tutorial**: [https://github.com/yunjey/pytorch-tutorial](https://github.com/yunjey/pytorch-tutorial)
1926

2027
## Getting Started
28+
29+
### Requirements
30+
31+
1. [C++](http://www.cplusplus.com/doc/tutorial/introduction/)
32+
2. [CMake](https://cmake.org/download/)
33+
3. [LibTorch v1.4.0](https://pytorch.org/cppdocs/installing.html)
34+
2135
### Fork/Clone and Build
2236

2337
```bash
2438
git clone https://github.com/prabhuomkar/pytorch-cpp.git
2539
cd pytorch-cpp
2640
```
41+
2742
#### Generate build system
43+
2844
```cmake
2945
cmake -B build #<options>
3046
```
3147
> **_Note for Windows users:_**<br>
32-
>Libtorch only supports 64bit Windows and an x64 generator needs to be specified. For Visual Studio this can be done by appending `-A x64` to the above command.
48+
> Libtorch only supports 64bit Windows and an x64 generator needs to be specified. For Visual Studio this can be done by appending `-A x64` to the above command.
3349
3450
Some useful options:
3551

@@ -40,9 +56,11 @@ Some useful options:
4056
| `-D CMAKE_PREFIX_PATH=path/to/libtorch/share/cmake/Torch` | | Skip the downloading of libtorch and use your own local version instead. |
4157

4258
#### Build
59+
4360
```cmake
4461
cmake --build build
4562
```
63+
4664
>**_Note for Windows users:_** <br>
4765
>The CMake script downloads the *Release* version of libtorch, so `--config Release` has to be appended to the build command.
4866
>
@@ -68,12 +86,4 @@ cmake --build build
6886
* [Generative Adversarial Networks](tutorials/advanced/generative_adversarial_network/main.cpp)
6987
* [Variational Auto-Encoder](tutorials/advanced/variational_autoencoder/src/main.cpp)
7088
* [Neural Style Transfer](tutorials/advanced/neural_style_transfer/src/main.cpp)
71-
* [Image Captioning (CNN-RNN)]()
72-
73-
## Dependencies
74-
- C++
75-
- PyTorch C++ API
76-
77-
## Authors
78-
- Omkar Prabhu - [prabhuomkar](https://github.com/prabhuomkar)
79-
- Markus Fleischhacker - [mfl28](https://github.com/mfl28)
89+
* [Image Captioning (CNN-RNN)]()

images/pytorch_cpp.png

97.9 KB
Loading

0 commit comments

Comments
 (0)