
<SYSTEM_TASK:>
Given the following text problem statement, write Python code to implement the functionality described below in problem statement
<END_TASK>
<USER_TASK:>
Problem:
Dataloader Arguments & Test/Train Dataloaders

<ASSISTANT_TASK:>
Python Code::
cuda = torch.cuda.is_available()
print("CUDA Available?", cuda)
SEED = 1
torch.manual_seed(SEED)
if cuda:
	torch.cuda.manual_seed(SEED)
dataloader_args = dict(shuffle=True, batch_size=128, num_workers=4, pin_memory=True) if cuda else dict(shuffle=True, batch_size=64)
train_loader = torch.utils.data.DataLoader(train, **dataloader_args)
test_loader = torch.utils.data.DataLoader(test, **dataloader_args)

<END_TASK>
