Commit 97713b54 authored by Amelie Royer's avatar Amelie Royer 🐼

Minor correction in MNISTm and SVHN Loaders

parent 9b999c8b
......@@ -54,5 +54,5 @@ class MNISTMConverter(Converter):
print()
MNISTMLoader = MNISTLoader
MNISTMLoader.shape = (32, 32, 3)
class MNISTMLoader(MNISTLoader):
shape = (32, 32, 3)
\ No newline at end of file
......@@ -51,5 +51,5 @@ class SVHNConverter(Converter):
print()
SVHNLoader = MNISTLoader
SVHNLoader.shape = (32, 32, 3)
\ No newline at end of file
class SVHNLoader(MNISTLoader):
shape = (32, 32, 3)
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment