Update train.py

train.py

@@ -67,7 +67,6 @@ class TensorMapperImageClassification(TensorMapper):
         in_channels=3,
         num_classes=10,     
         d_model = 256,
-       
         num_layers=4,
        
         
@@ -160,7 +159,7 @@ def test(dataloader, model, loss_fn):
 
 
 
-logname = "/home/abdullah/Desktop/Tensor_Mapper/Experiments_cifar10/logs_tensormapper/logs_cifar10.csv"
+logname = "/PATH/Tensor_Mapper/Experiments_cifar10/logs_tensormapper/logs_cifar10.csv"
 if not os.path.exists(logname):
   with open(logname, 'w') as logfile:
     logwriter = csv.writer(logfile, delimiter=',')
@@ -181,7 +180,7 @@ print("Done!")
 
 
 
-path = "/home/abdullah/Desktop/Tensor_Mapper/Experiments_cifar10/weights_tensormapper"
+path = "/PATH/Tensor_Mapper/Experiments_cifar10/weights_tensormapper"
 model_name = "TensorMapperImageClassification_cifar10"
 torch.save(model.state_dict(), f"{path}/{model_name}.pth")
 print(f"Saved Model State to {path}/{model_name}.pth ")