| 
									
										
										
										
											2023-11-29 08:17:26 -08:00
										 |  |  | # MNIST
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-12-11 20:45:39 -06:00
										 |  |  | This example shows how to run some simple models on MNIST.  | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | Install the dependencies: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | ``` | 
					
						
							|  |  |  | pip install -r requirements.txt | 
					
						
							|  |  |  | ``` | 
					
						
							| 
									
										
										
										
											2023-11-29 08:17:26 -08:00
										 |  |  | 
 | 
					
						
							|  |  |  | Run the example with: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | ``` | 
					
						
							|  |  |  | python main.py | 
					
						
							|  |  |  | ``` | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-12-23 16:34:45 +01:00
										 |  |  | By default, the example runs on the CPU. To run on the GPU, use:  | 
					
						
							| 
									
										
										
										
											2023-11-29 08:17:26 -08:00
										 |  |  | 
 | 
					
						
							|  |  |  | ``` | 
					
						
							|  |  |  | python main.py --gpu | 
					
						
							|  |  |  | ``` | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-12-23 16:34:45 +01:00
										 |  |  | For a full list of options run: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | ``` | 
					
						
							|  |  |  | python main.py --help | 
					
						
							|  |  |  | ``` | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-12-24 19:49:53 +05:30
										 |  |  | To run the PyTorch or JAX examples install the respective framework. |