Skip to content

Commit b16c326

Browse files
authored
(#27) Add a short demo of the learning engine
1 parent 68f0f6d commit b16c326

File tree

1 file changed

+3
-10
lines changed

1 file changed

+3
-10
lines changed

README.md

Lines changed: 3 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -21,16 +21,9 @@ Pocket is a fairly lightweight library built on the pupular [PyTorch](https://py
2121

2222
## Key Features
2323

24-
Pocket provides a range of engine classes that can perform training and testing with minimum amount of code. The following is a conceptual example.
25-
```python
26-
model = ...
27-
criterion = ...
28-
dataloader = ...
29-
# Instantiate the learning engine
30-
engine = Engine(model, criterion, dataloader)
31-
# Run the engine for a few epochs
32-
engine(10)
33-
```
24+
Pocket provides a range of engine classes that can perform training and testing with minimum amount of code. The following is a simple demo.
25+
26+
![pocket_demo](https://user-images.githubusercontent.com/11484831/102730290-56742e80-4388-11eb-85a7-c1ad67884ba8.gif)
3427

3528
Pocket provides two base classes of engines: __pocket.core.LearningEngine__ and __pocket.core.DistributedLearningEngine__ with the following features:
3629
- [x] CPU/GPU training

0 commit comments

Comments
 (0)