update

Author: kchu25

Project.toml

@@ -1,7 +1,7 @@
 name = "AutoComputationalGraphTuning"
 uuid = "9895cdd9-2e9e-4374-a2b7-90eb9a5a3bcd"
 authors = ["Shane Kuei-Hsien Chu (skchu@wustl.edu)"]
-version = "0.1.1"
+version = "0.1.2"
 
 [deps]
 CSV = "336ed68f-0bac-5ca0-87d4-7b16caf5d00b"

src/final_and_code/_helpers.jl

@@ -21,7 +21,7 @@ end
 """Create dataloaders for final model training. Returns: (dl_train, dl_test)"""
 function _create_final_dataloaders(setup, batch_size, seed)
     dl_train = Flux.DataLoader((setup.processed_data.train.tensor, setup.processed_data.train.labels),
-                               batchsize=batch_size, shuffle=true, partial=false, rng=MersenneTwister(seed))
+                               batchsize=batch_size, shuffle=true, partial=true, rng=MersenneTwister(seed))
     dl_test = Flux.DataLoader((setup.processed_data.test.tensor, setup.processed_data.test.labels),
                               batchsize=batch_size, shuffle=false, partial=true)
     return dl_train, dl_test