Repository navigation

deeplearning_framework_pl

Public

About

No description, website, or topics provided.

1 star

1 watching

Report repository

Name		Name	Last commit message	Last commit date
__pycache__		__pycache__
configs		configs
dataloader		dataloader
loss		loss
models		models
optimizer		optimizer
scheduler		scheduler
.gitignore		.gitignore
README.md		README.md
engine.py		engine.py
main.py		main.py

Repository files navigation

deeplearning_framework_pl

Introduction

Getting Started

1. train

CUDA_VISIBLE_DEVICES=0 python main.py --config ./configs/mnist.yaml --mode train

2. test

CUDA_VISIBLE_DEVICES=0 python main.py --config ./configs/mnist.yaml --mode test

About

No description, website, or topics provided.

1 star

1 watching

Report repository

Releases

Packages

Contributors

Languages