Implemented batchsize > 1 and Removed Matplotlib Dependency

Coldog2333 / pytoflow

The py version of toflow → https://github.com/anchen1011/toflow

MIT License

107 stars 31 forks source link

Implemented batchsize > 1 and Removed Matplotlib Dependency #14

Closed ahmadmughees closed 4 years ago

ahmadmughees commented 4 years ago

Previously the network was not able to handle the batch size greater than 1, which I have implemented using a simple logic. The repo had a dependency on a very old version of matplotlib(v2.0.2), which I removed and now it uses the PIL library which is installed by default with PyTorch.

ahmadmughees commented 4 years ago

there are few bugs in this. Don't Merge right now. Need to do some further testing and updations. Closing right now.