<filename>densenet/dataloader.py
import argparse

import torch
import torch.nn as nn
from torch import optim


if __name__ == '__main__':
    

<filename>densenet/dataloader.py
import argparse

import torch
import torch.nn as nn
from torch import optim


if __name__ == '__main__':
    
