File tree Expand file tree Collapse file tree 1 file changed +3
-10
lines changed Expand file tree Collapse file tree 1 file changed +3
-10
lines changed Original file line number Diff line number Diff 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
3528Pocket provides two base classes of engines: __ pocket.core.LearningEngine__ and __ pocket.core.DistributedLearningEngine__ with the following features:
3629- [x] CPU/GPU training
You can’t perform that action at this time.
0 commit comments