Yin Li
|
0d4ae3424e
|
Add cuda backend to inference
|
2021-05-30 23:04:50 -04:00 |
|
Yin Li
|
e20a3e3f62
|
Cleanup, Enable plt_power by default
|
2021-05-30 22:58:33 -04:00 |
|
Yin Li
|
8a3cd1843d
|
Move the executable script to the right place
so that calling it runs the local map2map
|
2021-05-18 13:29:13 -04:00 |
|
Yin Li
|
4be71a32d1
|
Add module and conda env recording
|
2021-03-18 15:27:48 -04:00 |
|
Yin Li
|
4799f8177c
|
Fix to rename --batches to --batch-size, former kept for BC
|
2020-09-12 18:30:14 -04:00 |
|
Yin Li
|
5ac9016987
|
Replace specific slurm scripts with general ones
|
2020-07-29 12:12:46 -07:00 |
|
Yin Li
|
e8744d6c7b
|
Remove adversary
|
2020-07-14 21:07:05 -04:00 |
|
Yin Li
|
818ed6923d
|
Add memmap to numpy data loading
|
2020-07-14 17:42:10 -04:00 |
|
Yin Li
|
c3d7456d0c
|
Change .pth to .pt following torch convention
|
2020-06-20 18:31:09 -04:00 |
|
Yin Li
|
996c0d3aed
|
Add auto loading of checkpoint
|
2020-04-21 18:34:18 -04:00 |
|
Yin Li
|
b67079bf72
|
Add runtime address and port determination and share them via file
Together with slurm step node counts, make it possible to launch
multiple training in one job
|
2020-02-13 19:56:54 -06:00 |
|
Yin Li
|
1dd2b6d748
|
Extend training epochs in example scripts
|
2020-02-09 17:32:27 -05:00 |
|
Yin Li
|
a237245514
|
Add displacement to density script
|
2020-01-31 11:13:55 -05:00 |
|
Yin Li
|
ec074748d0
|
Fix slurm script for training on 2 nodes on Frontera
|
2020-01-27 11:15:46 -05:00 |
|
Yin Li
|
84a369d4ed
|
Add super-resolution
|
2020-01-22 19:47:27 -05:00 |
|
Yin Li
|
c68b9928ee
|
Add separate input and target normalization
|
2020-01-21 18:51:32 -05:00 |
|
Yin Li
|
6e06682751
|
Update slurm script to include adversary model
|
2020-01-21 12:44:46 -05:00 |
|
Yin Li
|
9cf97b3ac1
|
Fix seeding bug introduced in the completely wrong commit f64b1e4
|
2020-01-06 20:20:05 -05:00 |
|
Yin Li
|
848dc87169
|
Change __dict__ to getattr
|
2019-12-23 16:11:43 -05:00 |
|
Yin Li
|
01b0c8b514
|
Add data division, good with data caching
|
2019-12-18 17:13:40 -05:00 |
|
Yin Li
|
de24f8d585
|
Fix testing glob patterns
|
2019-12-18 15:31:39 -05:00 |
|
Yin Li
|
843fe09a92
|
Add data caching, and new pad and crop features
|
2019-12-17 12:00:13 -05:00 |
|
Yin Li
|
7b6ff73be1
|
Move UNet with ResBlock to VNet and Revert UNet to the previous simple version
|
2019-12-09 21:53:27 -05:00 |
|
Yin Li
|
0764a1006e
|
Remove unnecessary arguments --in-channels and --out-channels
|
2019-12-09 10:19:21 -05:00 |
|
Yin Li
|
11c9caa1e2
|
Fix unstable training by limiting pytorch version to 1.1
|
2019-12-08 21:02:08 -05:00 |
|
Yin Li
|
afeefcaa9e
|
Merge branch 'master' of github.com:eelregit/map2map
Conflicts:
map2map/train.py
|
2019-12-03 17:52:01 -05:00 |
|
Yin Li
|
0211eed0ec
|
Add testing
|
2019-12-01 21:40:35 -05:00 |
|
Yin Li
|
88bfd11594
|
Add training
|
2019-11-30 16:31:10 -05:00 |
|