omarsar · August 31, 2019 14:12
diff --git a/test.py b/test.py
 ## test the model with 1 batch
 model = MyModel()
 for images, labels in trainloader:
    print("batch size:", images.shape)
    out = model(images)
    print(out.shape)
    break
	## test the model with 1 batch
	model = MyModel()
	for images, labels in trainloader:
	print("batch size:", images.shape)
	out = model(images)
	print(out.shape)
	break