Thanks for your reply. But I am still not able to figure out where to specify batch size.
In this ,         torch.nn.BatchNorm1d (num_features,eps=1e-05, momentum=0.1,affine=True,track_running_stats=True)     where to specify batch size as there is no provision for specifying that