recoilme commited on
Commit
b6d8e34
·
1 Parent(s): 29ec028
Files changed (3) hide show
  1. result_grid.jpg +2 -2
  2. src/sample.ipynb +0 -0
  3. train.py +3 -3
result_grid.jpg CHANGED

Git LFS Details

  • SHA256: 005e5ae9852b338a3afe6bc3d7a702c0756764247220e2f0a668f5006b5417a1
  • Pointer size: 132 Bytes
  • Size of remote file: 7.64 MB

Git LFS Details

  • SHA256: f52bf87a4ac6698eb77c01ff1af063274f13bd3db2691bfcaa92e25b3fd17d29
  • Pointer size: 132 Bytes
  • Size of remote file: 8.21 MB
src/sample.ipynb CHANGED
The diff for this file is too large to render. See raw diff
 
train.py CHANGED
@@ -28,9 +28,9 @@ from collections import deque
28
  # --------------------------- Параметры ---------------------------
29
  ds_path = "/workspace/sdxs3d/datasets/mjnj"
30
  project = "unet"
31
- batch_size = 96
32
- base_learning_rate = 6e-5
33
- min_learning_rate = 9e-6
34
  num_epochs = 50
35
  # samples/save per epoch
36
  sample_interval_share = 5
 
28
  # --------------------------- Параметры ---------------------------
29
  ds_path = "/workspace/sdxs3d/datasets/mjnj"
30
  project = "unet"
31
+ batch_size = 64
32
+ base_learning_rate = 4e-5
33
+ min_learning_rate = 8e-6
34
  num_epochs = 50
35
  # samples/save per epoch
36
  sample_interval_share = 5