Change deprecated cuda.amp to amp, update requirements.txt (#27)

Author: fdrose

week03_fast_pipelines/homework/task1/train.py

@@ -21,7 +21,7 @@ def train_epoch(
         images = images.to(device)
         labels = labels.to(device)
 
-        with torch.cuda.amp.autocast():
+        with torch.amp.autocast(device.type, dtype=torch.float16):
             outputs = model(images)
             loss = criterion(outputs, labels)
         # TODO: your code for loss scaling here

week03_fast_pipelines/seminar/requirements.txt

@@ -8,7 +8,7 @@ pandas==1.5.3
 py-spy==0.3.14
 torch==2.3.0
 torchtext
-torchvision==0.17.0
+torchvision==0.18.0
 tqdm==4.64.1
 vit_pytorch==0.40.2
 matplotlib==3.8.2